Browse Source

add onnx model parse api

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

+ 1
- 1
parser/caffe/caffe_parser.h View File

@@ -55,7 +55,7 @@ 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;
return domi::SUCCESS;
}

/**


+ 1
- 1
parser/onnx/onnx_parser.h View File

@@ -40,7 +40,7 @@ class OnnxModelParser : public domi::ModelParser {

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;
return domi::SUCCESS;
}

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


+ 1
- 1
parser/tensorflow/tensorflow_parser.h View File

@@ -92,7 +92,7 @@ class TensorFlowModelParser : public domi::ModelParser {

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;
return domi::SUCCESS;
}

/**


Loading…
Cancel
Save