Browse Source

!8158 fix onnx parser

Merge pull request !8158 from yankai10/merge_1103
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
8731d8f12e
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/lite/tools/converter/parser/onnx/onnx_arithmetic_operation_parser.cc

+ 2
- 2
mindspore/lite/tools/converter/parser/onnx/onnx_arithmetic_operation_parser.cc View File

@@ -234,13 +234,13 @@ STATUS OnnxMinParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::Node
return RET_NULL_PTR;
}

std::unique_ptr<schema::MinT> attr = std::make_unique<schema::MinT>();
std::unique_ptr<schema::MinimumT> attr = std::make_unique<schema::MinimumT>();
if (attr == nullptr) {
MS_LOG(ERROR) << "new op failed";
return RET_NULL_PTR;
}

op->primitive->value.type = schema::PrimitiveType_Min;
op->primitive->value.type = schema::PrimitiveType_Minimum;
op->primitive->value.value = attr.release();
return RET_OK;
}


Loading…
Cancel
Save