| /external/pytorch/torch/ao/quantization/ |
| D | quantize_jit.py | 43 def script_qconfig_dict(qconfig_dict): argument
68 def _prepare_jit(model, qconfig_dict, inplace=False, quant_type=QuantType.STATIC): argument
87 qconfig_dict, argument
109 def prepare_jit(model, qconfig_dict, inplace=False): argument
114 def prepare_dynamic_jit(model, qconfig_dict, inplace=False): argument
120 model, qconfig_dict, method_name="forward", inplace=False argument
211 model, qconfig_dict, method_name, inplace=False argument
220 qconfig_dict, argument
248 def quantize_jit(model, qconfig_dict, run_fn, run_args, inplace=False, debug=False): argument
310 def quantize_dynamic_jit(model, qconfig_dict, inplace=False, debug=False): argument
[all …]
|
| D | quantize.py | 73 qconfig_dict, argument
120 def propagate_qconfig_(module, qconfig_dict=None, prepare_custom_config_dict=None): argument
|
| /external/pytorch/test/ao/sparsity/ |
| D | test_kernels.py | 131 qconfig_dict, argument
|
| /external/pytorch/test/quantization/jit/ |
| D | test_ondevice_quantization.py | 58 def insert_observers(model, qconfig_dict): argument
65 def ptq_dynamic_quantize(model, qconfig_dict): argument
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | insert_observers.cpp | 49 const QConfigDict& qconfig_dict, in fillQConfigMap()
1652 const QConfigDict& qconfig_dict, in InsertObservers()
1679 const QConfigDict& qconfig_dict, in InsertObserversForOnDevicePTQ()
|
| /external/pytorch/test/quantization/fx/ |
| D | test_numeric_suite_fx.py | 833 self, m, example_inputs, results_len=0, qconfig_dict=None, prepare_fn=prepare_fx argument
862 qconfig_dict=None, argument
926 should_log_inputs=False, qconfig_dict=None, skip_scripting=False, argument
|
| D | test_quantize_fx.py | 5758 def _test(prepare_fn, qconfig_dict): argument
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | init.cpp | 337 const py::dict& qconfig_dict, in initJITBindings()
339 int quant_type_int) { in initJITBindings()
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantization.py | 235 def prepare_dynamic(model, qconfig_dict=None): argument
|
| /external/pytorch/test/dynamo/ |
| D | test_misc.py | 135 qconfig_dict = {"object_type": [(torch.nn.Linear, uniform_qconfig_8bit)]} variable
|