|
|
|
@ -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());
|
|
|
|
|