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) option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE)


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


include(cmake/external_libs/protobuf_shared.cmake) include(cmake/external_libs/protobuf_shared.cmake)
include(cmake/external_libs/protoc.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 Parse(const char *file, ge::Graph &graph) override;
Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &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 * @ingroup domi_omg


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

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


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


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

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


LOCAL_STATIC_LIBRARIES += libmmpa 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; ge::DataType ConvertToGeDataType(const uint32_t type) override;


Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; } 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 { Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override {
return domi::SUCCESS; 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 = 0, means empty tensor.
// dim i = -1 or -2, means unknown shape. // dim i = -1 or -2, means unknown shape.
GE_CHK_BOOL_RET_STATUS(user_input_dims_v[i] >= kValidShapeMinValue, domi::PARAM_INVALID, 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; 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 Parse(const char *file, ge::Graph &graph) override;


Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &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 * @ingroup domi_omg


Loading…
Cancel
Save