Browse Source

add onnx model parse api

pull/571/head
baker 4 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

@@ -66,6 +66,17 @@ class ModelParser {


/** /**
* @ingroup domi_omg * @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 * @brief Analyze network model data
* @param [in] proto network model * @param [in] proto network model
* @param [in|out] graph Save the network information after analysis * @param [in|out] graph Save the network information after analysis


Loading…
Cancel
Save