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
backend.trt.test_loader.TestModifyNetwork Class Reference
Inheritance diagram for backend.trt.test_loader.TestModifyNetwork:
Collaboration diagram for backend.trt.test_loader.TestModifyNetwork:

Public Member Functions

def test_mark_layerwise (self, identity_identity_network)
 
def test_mark_custom_outputs (self, identity_identity_network)
 
def test_exclude_outputs_with_mark_layerwise (self, identity_identity_network)
 
def test_mark_shape_outputs (self, reshape_network)
 
def test_unmark_shape_outputs (self, reshape_network)
 

Static Public Attributes

 reason
 

Member Function Documentation

◆ test_mark_layerwise()

def backend.trt.test_loader.TestModifyNetwork.test_mark_layerwise (   self,
  identity_identity_network 
)

◆ test_mark_custom_outputs()

def backend.trt.test_loader.TestModifyNetwork.test_mark_custom_outputs (   self,
  identity_identity_network 
)

◆ test_exclude_outputs_with_mark_layerwise()

def backend.trt.test_loader.TestModifyNetwork.test_exclude_outputs_with_mark_layerwise (   self,
  identity_identity_network 
)

◆ test_mark_shape_outputs()

def backend.trt.test_loader.TestModifyNetwork.test_mark_shape_outputs (   self,
  reshape_network 
)

◆ test_unmark_shape_outputs()

def backend.trt.test_loader.TestModifyNetwork.test_unmark_shape_outputs (   self,
  reshape_network 
)

Member Data Documentation

◆ reason

backend.trt.test_loader.TestModifyNetwork.reason
static

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