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.tensor_quantizer_test.TestTensorQuantizer Class Reference

Public Member Functions

def test_simple_run (self)
 
def test_simple_run_no_fake (self)
 
def test_per_tensor_scale (self)
 
def test_per_channel_scale (self, verbose)
 
def test_learn_amax (self)
 
def test_clip_mode (self)
 
def test_scale_amax (self)
 
def test_disable (self)
 
def test_state_loading (self)
 
def test_properties (self)
 
def test_init_calib (self)
 
def test_max_calib (self)
 
def test_entropy_and_percentile_calib (self)
 
def test_setters (self)
 

Member Function Documentation

◆ test_simple_run()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_simple_run (   self)
Quantizer calls fake_tensor_quant by default

◆ test_simple_run_no_fake()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_simple_run_no_fake (   self)
Quantizer fake_quant=False calls tensor_quant and sets the scale property

◆ test_per_tensor_scale()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_per_tensor_scale (   self)
Quantizer performs expected quantization

◆ test_per_channel_scale()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_per_channel_scale (   self,
  verbose 
)
Quantizer performs per channel scaling

◆ test_learn_amax()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_learn_amax (   self)
Test the clip implied by learn_amax

◆ test_clip_mode()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_clip_mode (   self)
Test the clip stage only

◆ test_scale_amax()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_scale_amax (   self)

◆ test_disable()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_disable (   self)

◆ test_state_loading()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_state_loading (   self)
Test quant_desc loading via state_dict

◆ test_properties()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_properties (   self)

◆ test_init_calib()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_init_calib (   self)

◆ test_max_calib()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_max_calib (   self)

◆ test_entropy_and_percentile_calib()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_entropy_and_percentile_calib (   self)
Don't really have a good way to test it.

◆ test_setters()

def tests.tensor_quantizer_test.TestTensorQuantizer.test_setters (   self)

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