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
tests.integration_test.TestNetwork Class Reference

Public Member Functions

def test_simple_build (self)
 
def test_forward (self)
 
def test_backward (self)
 
def test_amp_fp16 (self)
 
def test_quant_module_replacement (self)
 
def test_calibration (self)
 

Detailed Description

test basic operations of quantized network

Member Function Documentation

◆ test_simple_build()

def tests.integration_test.TestNetwork.test_simple_build (   self)
test instantiation

◆ test_forward()

def tests.integration_test.TestNetwork.test_forward (   self)
test forward pass with random data

◆ test_backward()

def tests.integration_test.TestNetwork.test_backward (   self)
test one iteration with random data and labels

◆ test_amp_fp16()

def tests.integration_test.TestNetwork.test_amp_fp16 (   self)
test one iteration with random data and labels

◆ test_quant_module_replacement()

def tests.integration_test.TestNetwork.test_quant_module_replacement (   self)
test monkey patching of modules with their quantized versions

◆ test_calibration()

def tests.integration_test.TestNetwork.test_calibration (   self)

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