Browse Source

add onnx model parse api

pull/571/head
baker 5 years ago
parent
commit
93600779f7
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      inc/framework/omg/parser/model_parser.h

+ 11
- 0
inc/framework/omg/parser/model_parser.h View File

@@ -64,6 +64,17 @@ class ModelParser {
*/
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) = 0;

/**
* @ingroup domi_omg
* @brief Parse relevant data from memory and save it to graph
* @param [in] input Model file memory data and 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;

/**
* @ingroup domi_omg
* @brief Analyze network model data


Loading…
Cancel
Save