Variables | |
graph = gs.import_onnx(onnx.load("model.onnx")) | |
list | fake_node = [node for node in graph.nodes if node.op == "FakeNodeToRemove"][0] |
list | inp_node = fake_node.i() |
outputs | |
remove.graph = gs.import_onnx(onnx.load("model.onnx")) |
list remove.fake_node = [node for node in graph.nodes if node.op == "FakeNodeToRemove"][0] |
list remove.inp_node = fake_node.i() |
remove.outputs |