Browse Source

add parse onnx model api

pull/131/head
baker 5 years ago
parent
commit
635e52e91f
3 changed files with 9 additions and 0 deletions
  1. +3
    -0
      parser/caffe/caffe_parser.h
  2. +3
    -0
      parser/onnx/onnx_parser.h
  3. +3
    -0
      parser/tensorflow/tensorflow_parser.h

+ 3
- 0
parser/caffe/caffe_parser.h View File

@@ -54,6 +54,9 @@ class CaffeModelParser : public domi::ModelParser {
*/
Status Parse(const char *file, ge::Graph &graph) override;
Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override;
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) {
return domi::SUCCESS;
}

/**
* @ingroup domi_omg


+ 3
- 0
parser/onnx/onnx_parser.h View File

@@ -39,6 +39,9 @@ class OnnxModelParser : public domi::ModelParser {
ge::DataType ConvertToGeDataType(const uint32_t type) override;

Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; }
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) {
return domi::SUCCESS;
}

Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override {
return domi::SUCCESS;


+ 3
- 0
parser/tensorflow/tensorflow_parser.h View File

@@ -91,6 +91,9 @@ class TensorFlowModelParser : public domi::ModelParser {
Status Parse(const char *file, ge::Graph &graph) override;

Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override;
virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) {
return domi::SUCCESS;
}

/**
* @ingroup domi_omg


Loading…
Cancel
Save