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
tools.test_surgeon.TestSurgeonInsert Class Reference
Inheritance diagram for tools.test_surgeon.TestSurgeonInsert:
Collaboration diagram for tools.test_surgeon.TestSurgeonInsert:

Public Member Functions

def check_insert_model (self, path, expected_node_ops, expected_graph_input_names, expected_graph_output_names)
 
def test_input_is_output (self)
 
def test_graph_output (self)
 
def test_at_graph_input (self)
 

Member Function Documentation

◆ check_insert_model()

def tools.test_surgeon.TestSurgeonInsert.check_insert_model (   self,
  path,
  expected_node_ops,
  expected_graph_input_names,
  expected_graph_output_names 
)
Here is the caller graph for this function:

◆ test_input_is_output()

def tools.test_surgeon.TestSurgeonInsert.test_input_is_output (   self)
Here is the call graph for this function:

◆ test_graph_output()

def tools.test_surgeon.TestSurgeonInsert.test_graph_output (   self)
Here is the call graph for this function:

◆ test_at_graph_input()

def tools.test_surgeon.TestSurgeonInsert.test_at_graph_input (   self)
Here is the call graph for this function:

The documentation for this class was generated from the following file: