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) |
test basic operations of quantized network
def tests.integration_test.TestNetwork.test_simple_build | ( | self | ) |
test instantiation
def tests.integration_test.TestNetwork.test_forward | ( | self | ) |
test forward pass with random data
def tests.integration_test.TestNetwork.test_backward | ( | self | ) |
test one iteration with random data and labels
def tests.integration_test.TestNetwork.test_amp_fp16 | ( | self | ) |
test one iteration with random data and labels
def tests.integration_test.TestNetwork.test_quant_module_replacement | ( | self | ) |
test monkey patching of modules with their quantized versions
def tests.integration_test.TestNetwork.test_calibration | ( | self | ) |