|
@@ -54,14 +54,14 @@ class ModelParser { |
|
|
virtual Status Parse(const char *file, ge::Graph &graph) = 0; |
|
|
virtual Status Parse(const char *file, ge::Graph &graph) = 0; |
|
|
|
|
|
|
|
|
/** |
|
|
/** |
|
|
* @ingroup domi_omg |
|
|
|
|
|
* @brief Parse relevant data from memory and save it to graph |
|
|
|
|
|
* @param [in] input Model file memory data |
|
|
|
|
|
* @param [in|out] graph A graph for saving the model information after analysis |
|
|
|
|
|
* @return SUCCESS |
|
|
|
|
|
* @return FAILED |
|
|
|
|
|
* @author |
|
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
* @ingroup domi_omg |
|
|
|
|
|
* @brief Parse relevant data from memory and save it to graph |
|
|
|
|
|
* @param [in] input Model file memory data |
|
|
|
|
|
* @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::ComputeGraphPtr &graph) = 0; |
|
|
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) = 0; |
|
|
|
|
|
|
|
|
/** |
|
|
/** |
|
@@ -107,12 +107,12 @@ class ModelParser { |
|
|
*/ |
|
|
*/ |
|
|
virtual Status ToJson(const char *model_file, const char *json_file) { return domi::SUCCESS; } |
|
|
virtual Status ToJson(const char *model_file, const char *json_file) { return domi::SUCCESS; } |
|
|
|
|
|
|
|
|
/* |
|
|
|
|
|
* @ingroup domi_omg |
|
|
|
|
|
* @brief Convert network data type |
|
|
|
|
|
* @param [in] type Data type to be converted |
|
|
|
|
|
* @return ge::DataType |
|
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
/** |
|
|
|
|
|
* @ingroup domi_omg |
|
|
|
|
|
* @brief Convert network data type |
|
|
|
|
|
* @param [in] type Data type to be converted |
|
|
|
|
|
* @return ge::DataType |
|
|
|
|
|
*/ |
|
|
virtual ge::DataType ConvertToGeDataType(const uint32_t type) = 0; |
|
|
virtual ge::DataType ConvertToGeDataType(const uint32_t type) = 0; |
|
|
|
|
|
|
|
|
virtual Status ParseAllGraph(const google::protobuf::Message *root_proto, ge::ComputeGraphPtr &root_graph) = 0; |
|
|
virtual Status ParseAllGraph(const google::protobuf::Message *root_proto, ge::ComputeGraphPtr &root_graph) = 0; |
|
|