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}
