add error info

pull/14504/head
liuyang_655 4 years ago
parent 4607b12476
commit 8a3c74a722

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