!643 add onnx model parse api

From: @chen-hua-baker
Reviewed-by: @xchu42,@xchu42,@wqtshg
Signed-off-by: @wqtshg
pull/643/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 8c890d5cde

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

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