Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_lite_script_module.py506 quantized_model = self._create_quantized_model(
513 quantized_model = self._create_quantized_model(model_class=TwoLayerLinearModel)
518 quantized_model = self._create_quantized_model(
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py1182 def _create_quantized_model(self, model_class: Type[torch.nn.Module], **kwargs): member in QuantizationLiteTestCase