Searched refs:_create_quantized_model (Results 1 – 2 of 2) sorted by relevance
506 quantized_model = self._create_quantized_model(513 quantized_model = self._create_quantized_model(model_class=TwoLayerLinearModel)518 quantized_model = self._create_quantized_model(
1182 def _create_quantized_model(self, model_class: Type[torch.nn.Module], **kwargs): member in QuantizationLiteTestCase