Searched refs:checkDynamicQuantizedLinear (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_quantize_eager_ptq.py | 1103 self.checkDynamicQuantizedLinear(model.fc1, dtype) 1143 self.checkDynamicQuantizedLinear(model.fc2, dtype=dtype) 1174 self.checkDynamicQuantizedLinear(model.fc3, dtype=dtype) 1175 self.checkDynamicQuantizedLinear(model.sub2.fc1, dtype=dtype) 1207 self.checkDynamicQuantizedLinear(model.sub2.fc1, dtype=dtype) 1208 self.checkDynamicQuantizedLinear(model.sub2.fc2, dtype=dtype) 1209 self.checkDynamicQuantizedLinear(model.fc3, dtype=dtype) 1240 self.checkDynamicQuantizedLinear(model.sub2.fc1, dtype=dtype) 1241 self.checkDynamicQuantizedLinear(model.sub2.fc2, dtype=dtype) 1242 self.checkDynamicQuantizedLinear(model.fc3, dtype=dtype) [all …]
|
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_numeric_suite_fx.py | 2419 self.checkDynamicQuantizedLinear(msq.shadow_wrapper_0_2.mod_0, torch.qint8) 2448 self.checkDynamicQuantizedLinear(msq.shadow_wrapper_0_2.mod_0, torch.qint8) 2479 self.checkDynamicQuantizedLinear(msq.shadow_wrapper_0_2.mod_0, torch.qint8) 2480 self.checkDynamicQuantizedLinear(msq.shadow_wrapper_1_1.mod_0, torch.qint8)
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_quantization.py | 640 def checkDynamicQuantizedLinear(self, mod, dtype): member in QuantizationTestCase
|