add onnx model parse api

pull/637/head
baker 4 years ago
parent 1f05cd0b5c
commit 984fd1bae6

@ -57,6 +57,7 @@ class ModelParser {
* @ingroup domi_omg * @ingroup domi_omg
* @brief Parse relevant data from memory and save it to graph * @brief Parse relevant data from memory and save it to graph
* @param [in] input Model file memory data * @param [in] input Model file memory data
* @param [in] input Model file memory size
* @param [in|out] graph A graph for saving the model information after analysis * @param [in|out] graph A graph for saving the model information after analysis
* @return SUCCESS * @return SUCCESS
* @return FAILED * @return FAILED
@ -64,6 +65,20 @@ class ModelParser {
*/ */
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) = 0; virtual Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) = 0;
#ifndef ONLY_COMPILE_OPEN_SRC
/**
* @ingroup domi_omg
* @brief Parse relevant data from memory and save it to graph
* @param [in] input Model file memory data
* @param [in] input Model file memory size
* @param [in|out] graph A graph for saving the model information after analysis
* @return SUCCESS
* @return FAILED
* @author
*/
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) = 0;
#endif
/** /**
* @ingroup domi_omg * @ingroup domi_omg
* @brief Analyze network model data * @brief Analyze network model data
@ -83,8 +98,7 @@ class ModelParser {
* @return SUCCESS * @return SUCCESS
* @return Others failed * @return Others failed
*/ */
virtual Status ParseProtoWithSubgraph(const google::protobuf::Message *proto, virtual Status ParseProtoWithSubgraph(const google::protobuf::Message *proto, GetGraphCallback callback,
GetGraphCallback callback,
ge::ComputeGraphPtr &graph) = 0; ge::ComputeGraphPtr &graph) = 0;
/** /**
* @ingroup domi_omg * @ingroup domi_omg

Loading…
Cancel
Save