Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dcustom_config.py120 def set_float_to_observed_mapping( member in PrepareCustomConfig
300 conf.set_float_to_observed_mapping(
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py2691 …prepare_custom_config.set_float_to_observed_mapping(self._DummyFloatModule, self._DummyObservedMod…
2800 .set_float_to_observed_mapping(self._DummyFloatModule, self._DummyObservedModule) \
4772 .set_float_to_observed_mapping(torch.nn.LSTM, torch.ao.nn.quantizable.LSTM)
4958 .set_float_to_observed_mapping(torch.nn.LSTM, UserObservedLSTM)