Browse Source

ONLY_COMPILE_OPEN_SRC

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

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

@@ -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


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

@@ -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;
}


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

@@ -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


Loading…
Cancel
Save