Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dcustom_config.py167 def set_output_quantized_indexes(self, indexes: List[int]) -> PrepareCustomConfig: member in PrepareCustomConfig
312 conf.set_output_quantized_indexes(
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py2721 prepare_custom_config.set_output_quantized_indexes([0, 1])
2804 .set_output_quantized_indexes([0, 1]) \