From bcc4ab7853d492c2dfbf290ecb486829fffb19c7 Mon Sep 17 00:00:00 2001 From: baker Date: Sat, 12 Dec 2020 16:51:19 +0800 Subject: [PATCH] ONLY_COMPILE_OPEN_SRC --- parser/caffe/caffe_parser.h | 3 ++- parser/onnx/onnx_parser.h | 3 ++- parser/tensorflow/tensorflow_parser.h | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/parser/caffe/caffe_parser.h b/parser/caffe/caffe_parser.h index 3818067..4e0d74b 100644 --- a/parser/caffe/caffe_parser.h +++ b/parser/caffe/caffe_parser.h @@ -54,10 +54,11 @@ 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; +#ifndef ONLY_COMPILE_OPEN_SRC virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { return domi::SUCCESS; } - +#endif /** * @ingroup domi_omg * @brief Convert model files to JSON format diff --git a/parser/onnx/onnx_parser.h b/parser/onnx/onnx_parser.h index 5eba094..1764c3f 100644 --- a/parser/onnx/onnx_parser.h +++ b/parser/onnx/onnx_parser.h @@ -39,10 +39,11 @@ 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; } +#ifndef ONLY_COMPILE_OPEN_SRC virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { return domi::SUCCESS; } - +#endif 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 c2eee31..b1d3a58 100644 --- a/parser/tensorflow/tensorflow_parser.h +++ b/parser/tensorflow/tensorflow_parser.h @@ -91,10 +91,11 @@ 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; +#ifndef ONLY_COMPILE_OPEN_SRC virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { return domi::SUCCESS; } - +#endif /** * @ingroup domi_omg * @brief Convert model files to JSON format