Searched refs:checkDynamicQuantizedModule (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 838 …self.checkDynamicQuantizedModule(quantized_model.emb, torch.ao.nn.quantized.EmbeddingBag, torch.qu…
859 …self.checkDynamicQuantizedModule(quantized_model.emb, torch.ao.nn.quantized.EmbeddingBag, torch.qu…
1351 self.checkDynamicQuantizedModule(model_quantized.mod, mod_type_map[module_type], dtype)
1429 … self.checkDynamicQuantizedModule(model_quantized.mod, mod_type_map[module_type], dtype)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 701 def checkDynamicQuantizedModule(self, mod, reference_module_type, dtype): member in QuantizationTestCase
|