def create_model(self): mp = ModelProto() mp.ir_version = ONNX_IR_VERSION op = mp.opset_import.add() op.domain = "" # empty string indicates ONNX domain op.version = ONNX_OPSET_VERSION # nn_opset = mp.opset_import.add() # nn_opset.domain = NNABLA_DOMAIN # nn_opset.version = NNABLA_OPSET_VERSION mp.producer_name = PRODUCER_NAME mp.producer_version = PRODUCER_VERSION mp.domain = NNABLA_DOMAIN self._model_proto = mp
def nnp_model_to_onnx_protobuf(nnp, batch_size): mp = ModelProto() mp.ir_version = ONNX_IR_VERSION op = mp.opset_import.add() op.domain = "" # empty string indicates ONNX domain op.version = ONNX_OPSET_VERSION # nn_opset = mp.opset_import.add() # nn_opset.domain = NNABLA_DOMAIN # nn_opset.version = NNABLA_OPSET_VERSION mp.producer_name = PRODUCER_NAME mp.producer_version = PRODUCER_VERSION mp.domain = NNABLA_DOMAIN nnp_model_to_onnx_graph(mp.graph, nnp, batch_size) return mp