def mk_model(opset_versions: List[Tuple[Text, int]]) -> ModelProto: graph = helper.make_graph([], "my graph", [], []) return helper.make_model_gen_version(graph, opset_imports=[ helper.make_opsetid(*pair) for pair in opset_versions ])
def reset_model(self, opset_ver=None): if opset_ver is not None: opset_imports = [make_opsetid("", opset_ver)] else: opset_imports = [self.opset_import] self.model = make_model_gen_version(onnx.GraphProto(), opset_imports=opset_imports) self.op_counter = collections.Counter()
def __init__(self, opset_ver=OPSET_VER): self.opset_import = make_opsetid("", opset_ver) self.model = make_model_gen_version(onnx.GraphProto(), opset_imports=[self.opset_import]) self.op_counter = collections.Counter() self.ctx = C.CheckerContext() self.ctx.ir_version = self.model.ir_version self.ctx.opset_imports = {'': opset_ver}