Home
last modified time | relevance | path

Searched refs:checkDynamicQuantizedLinear (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py1103 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 Dtest_numeric_suite_fx.py2419 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 Dcommon_quantization.py640 def checkDynamicQuantizedLinear(self, mod, dtype): member in QuantizationTestCase