Browse Source

add onnx model parse api

tags/v1.2.0
baker 4 years ago
parent
commit
30ae5dbb10
2 changed files with 4 additions and 1 deletions
  1. +3
    -0
      inc/framework/common/ge_types.h
  2. +1
    -1
      inc/framework/omg/parser/parser_inner_ctx.h

+ 3
- 0
inc/framework/common/ge_types.h View File

@@ -37,6 +37,9 @@ enum FrameworkType {
MINDSPORE = 1,
TENSORFLOW = 3,
ANDROID_NN,
#ifndef ONLY_COMPILE_OPEN_SRC
ONNX,
#endif
FRAMEWORK_RESERVED,
};



+ 1
- 1
inc/framework/omg/parser/parser_inner_ctx.h View File

@@ -59,7 +59,7 @@ struct ParserContext {
bool train_flag = false;
domi::domiTensorFormat_t format = domi::DOMI_TENSOR_ND;
domi::FrameworkType type = domi::FRAMEWORK_RESERVED;
RunMode run_mode = ONLY_PRE_CHECK;
RunMode run_mode = GEN_OM_MODEL;
// save caffe custom proto path, used by caffe parse
std::string custom_proto_path;
// save caffe proto path, used by caffe parse


Loading…
Cancel
Save