Home
last modified time | relevance | path

Searched refs:set_observed_to_quantized_mapping (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dcustom_config.py393 def set_observed_to_quantized_mapping( member in ConvertCustomConfig
443 conf.set_observed_to_quantized_mapping(
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py2814 convert_custom_config.set_observed_to_quantized_mapping(
2854 … .set_observed_to_quantized_mapping(self._DummyObservedModule, self._DummyQuantizedModule) \
4774 … .set_observed_to_quantized_mapping(torch.ao.nn.quantizable.LSTM, torch.ao.nn.quantized.LSTM)
4960 .set_observed_to_quantized_mapping(torch.ao.nn.quantizable.LSTM, UserQuantizedLSTM)