Searched refs:float_module_class (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | quantization_mappings.py | 239 float_module_class: Callable,
254 static_quant_module_class = all_mappings.get(float_module_class, None)
263 float_module_class: Callable,
274 dynamic_quant_module_class = all_mappings.get(float_module_class, None)
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 79 float_module_class, argument
86 self.conv = float_module_class(**extra_module_kwargs)
99 self.conv = float_module_class(**extra_module_kwargs)
274 self, float_module_class, quantized_module_class, extra_module_kwargs): argument
282 self.activation_op = float_module_class(**extra_module_kwargs)
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_module.py | 1152 …def _test_activation_module_impl(self, name, float_module_class, quantized_module_class, extra_kwa… argument
1168 float_mod = float_module_class(**extra_kwargs).float()
|