|
|
|
@ -57,6 +57,7 @@ class ModelParser {
|
|
|
|
|
* @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
|
|
|
|
@ -64,6 +65,20 @@ class ModelParser {
|
|
|
|
|
*/
|
|
|
|
|
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
|
|
|
|
|
* @brief Analyze network model data
|
|
|
|
@ -83,8 +98,7 @@ class ModelParser {
|
|
|
|
|
* @return SUCCESS
|
|
|
|
|
* @return Others failed
|
|
|
|
|
*/
|
|
|
|
|
virtual Status ParseProtoWithSubgraph(const google::protobuf::Message *proto,
|
|
|
|
|
GetGraphCallback callback,
|
|
|
|
|
virtual Status ParseProtoWithSubgraph(const google::protobuf::Message *proto, GetGraphCallback callback,
|
|
|
|
|
ge::ComputeGraphPtr &graph) = 0;
|
|
|
|
|
/**
|
|
|
|
|
* @ingroup domi_omg
|
|
|
|
|