TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
replace Namespace Reference

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"]]
 

Function Documentation

◆ replace_with_clip()

def replace.replace_with_clip (   self,
  inputs,
  outputs 
)

Variable Documentation

◆ graph

replace.graph = gs.import_onnx(onnx.load("model.onnx"))

◆ tmap

replace.tmap = graph.tensors()

◆ inputs

list replace.inputs = [tmap["identity_out_0"], tmap["onnx_graphsurgeon_constant_5"], tmap["onnx_graphsurgeon_constant_2"]]

◆ outputs

list replace.outputs = [tmap["max_out_6"]]