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
|