Browse Source

Merge branch 'development' of gitee.com:dajunli/parser into development

pull/138/head
dajunli 5 years ago
parent
commit
a5241e43a0
8 changed files with 14 additions and 4 deletions
  1. +1
    -1
      CMakeLists.txt
  2. +1
    -1
      metadef
  3. +3
    -0
      parser/caffe/caffe_parser.h
  4. +1
    -0
      parser/onnx/CMakeLists.txt
  5. +1
    -0
      parser/onnx/module.mk
  6. +3
    -0
      parser/onnx/onnx_parser.h
  7. +1
    -2
      parser/tensorflow/tensorflow_data_parser.cc
  8. +3
    -0
      parser/tensorflow/tensorflow_parser.h

+ 1
- 1
CMakeLists.txt View File

@@ -10,7 +10,7 @@ endif()
option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE)

if (ENABLE_OPEN_SRC)
set(HI_PYTHON python3.7)
set(HI_PYTHON python3)

include(cmake/external_libs/protobuf_shared.cmake)
include(cmake/external_libs/protoc.cmake)


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 220067ec17e13e5cc913e9e2c5895fcabaa5bd59
Subproject commit d19c9c5c92f21a0335c18681dcceed44f3a54ddc

+ 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


+ 1
- 0
parser/onnx/CMakeLists.txt View File

@@ -74,6 +74,7 @@ target_link_libraries(fmk_onnx_parser PRIVATE
json
-lrt
-ldl
error_manager
)

############ install ############


+ 1
- 0
parser/onnx/module.mk View File

@@ -50,6 +50,7 @@ LOCAL_SHARED_LIBRARIES := \
libparser_common \
libgraph \
libregister \
liberror_manager \

LOCAL_STATIC_LIBRARIES += libmmpa



+ 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;


+ 1
- 2
parser/tensorflow/tensorflow_data_parser.cc View File

@@ -139,8 +139,7 @@ Status TensorFlowDataParser::CheckInputShape(const std::string &name) {
// dim i = 0, means empty tensor.
// dim i = -1 or -2, means unknown shape.
GE_CHK_BOOL_RET_STATUS(user_input_dims_v[i] >= kValidShapeMinValue, domi::PARAM_INVALID,
"parse data node %s: shape contains placeholder ,but not designated by user",
name.c_str());
"parse data node %s: shape contains placeholder ,but not designated by user", name.c_str());
}
return 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