Functions | |
def | replace_with_clip (self, inputs, outputs) |
Variables | |
graph = gs.import_onnx(onnx.load("model.onnx")) | |
tmap = graph.tensors() | |
list | inputs = [tmap["identity_out_0"], tmap["onnx_graphsurgeon_constant_5"], tmap["onnx_graphsurgeon_constant_2"]] |
list | outputs = [tmap["max_out_6"]] |
def replace.replace_with_clip | ( | self, | |
inputs, | |||
outputs | |||
) |
replace.graph = gs.import_onnx(onnx.load("model.onnx")) |
replace.tmap = graph.tensors() |
list replace.inputs = [tmap["identity_out_0"], tmap["onnx_graphsurgeon_constant_5"], tmap["onnx_graphsurgeon_constant_2"]] |