From 635e52e91f8ebd1d90680b7836d141d14994f0e5 Mon Sep 17 00:00:00 2001 From: baker Date: Thu, 10 Dec 2020 09:24:41 +0800 Subject: [PATCH] add parse onnx model api --- parser/caffe/caffe_parser.h | 3 +++ parser/onnx/onnx_parser.h | 3 +++ parser/tensorflow/tensorflow_parser.h | 3 +++ 3 files changed, 9 insertions(+) diff --git a/parser/caffe/caffe_parser.h b/parser/caffe/caffe_parser.h index 9a252f9..d2e4701 100644 --- a/parser/caffe/caffe_parser.h +++ b/parser/caffe/caffe_parser.h @@ -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 diff --git a/parser/onnx/onnx_parser.h b/parser/onnx/onnx_parser.h index bdb93c5..7e4c3e2 100644 --- a/parser/onnx/onnx_parser.h +++ b/parser/onnx/onnx_parser.h @@ -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; diff --git a/parser/tensorflow/tensorflow_parser.h b/parser/tensorflow/tensorflow_parser.h index 2200a8d..8d58328 100644 --- a/parser/tensorflow/tensorflow_parser.h +++ b/parser/tensorflow/tensorflow_parser.h @@ -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