◆ 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_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: