!14504 Add onnx exporter error info

From: @liuyang_655
Reviewed-by: @zh_qh,@zhunaipan
Signed-off-by: @zh_qh
pull/14504/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit d00ad5f11b

@ -1025,12 +1025,11 @@ size_t OnnxExporter::ExportPrimitive(const FuncGraphPtr & /*func_graph*/, std::m
auto op_map = OpConvertRegistry::GetOpConvertMap();
auto op_iter = op_map.find(prim->name());
if (op_iter == op_map.end()) {
MS_LOG(EXCEPTION) << "Can not find key " << prim->name() << " in convert map";
MS_LOG(EXCEPTION) << "Can not find key " << prim->name() << " in convert map. "
<< "Exporting " << prim->name() << " operator is not yet supported.";
}
const OpNameInfo &op_convert_info = op_iter->second;
auto node_idx = AllocateNodeIndex();
onnx::NodeProto *node_proto = graph_proto->add_node();
node_proto->add_output(std::to_string(node_idx));
node_proto->set_op_type(op_convert_info.onnx_type());

Loading…
Cancel
Save