Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dcustom_config.py159 def set_input_quantized_indexes(self, indexes: List[int]) -> PrepareCustomConfig: member in PrepareCustomConfig
309 conf.set_input_quantized_indexes(
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py2715 prepare_custom_config.set_input_quantized_indexes([0, 1])
2803 .set_input_quantized_indexes([0, 1]) \