Home
last modified time | relevance | path

Searched refs:calculate_qparams (Results 1 – 25 of 48) sorted by relevance

12

/external/pytorch/test/quantization/core/experimental/
Dtest_nonuniform_observer.py18 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=False)
34 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=False)
76 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=False)
120 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=True)
170 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=False)
207 alpha, gamma, quantization_levels, level_indices = obs.calculate_qparams(signed=True)
Dtest_quantizer.py29 … alpha, gamma, quantization_levels, level_indices = apot_observer.calculate_qparams(signed=False)
41 scale, zero_point = uniform_observer.calculate_qparams()
77 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
117 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
162 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
210 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
Dtest_fake_quantize.py23 alpha, gamma, quantization_levels, level_indices = apot_fake.calculate_qparams(signed=False)
29 qparams_expected = observer.calculate_qparams(signed=False)
49 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
87 alpha, gamma, quantization_levels, level_indices = observer.calculate_qparams(signed=False)
Dtest_quantized_tensor.py20 qparams = observer.calculate_qparams(signed=False)
/external/pytorch/torch/ao/nn/quantized/modules/
Dactivation.py81 scale, zero_point = mod.activation_post_process.calculate_qparams()
113 scale, zero_point = mod.activation_post_process.calculate_qparams()
154 scale, zero_point = mod.activation_post_process.calculate_qparams()
185 ) = mod.activation_post_process.calculate_qparams()
220 scale, zero_point = mod.activation_post_process.calculate_qparams()
312 scale, zero_point = mod.activation_post_process.calculate_qparams()
321 wt_scale, wt_zp = observer.calculate_qparams()
338 wt_scale, wt_zp = observer.calculate_qparams()
Dnormalization.py63 scale, zero_point = mod.activation_post_process.calculate_qparams()
134 scale, zero_point = mod.activation_post_process.calculate_qparams()
190 scale, zero_point = mod.activation_post_process.calculate_qparams()
257 scale, zero_point = mod.activation_post_process.calculate_qparams()
324 scale, zero_point = mod.activation_post_process.calculate_qparams()
D__init__.py130 scale, zero_point = mod.activation_post_process.calculate_qparams()
Dbatchnorm.py23 scale, zero_point = activation_post_process.calculate_qparams()
/external/pytorch/torch/ao/quantization/experimental/
Dfake_quantize.py22 def calculate_qparams(self, signed=False): # type: ignore[override] member in APoTFakeQuantize
23 return self.activation_post_process.calculate_qparams(signed=signed)
28 result = self.activation_post_process.calculate_qparams(signed=False)
Dobserver.py37 def calculate_qparams(self, signed): member in APoTObserver
151 alpha, gamma, quantization_levels, level_indices = self.calculate_qparams(
Dadaround_fake_quantize.py64 def calculate_qparams(self) -> Tuple[torch.Tensor, torch.Tensor]: member in AdaroundFakeQuantizer
120 _scale, _zero_point = self.activation_post_process.calculate_qparams()
Dlinear.py48 ) = observer.calculate_qparams(signed=False)
/external/pytorch/test/quantization/core/
Dtest_workflow_module.py116 qparams = myobs.calculate_qparams()
132 qparams = myobs.calculate_qparams()
147 loaded_qparams = loaded_obs.calculate_qparams()
150 self.assertEqual(myobs.calculate_qparams(), loaded_obs.calculate_qparams())
174 qparams = myobs.calculate_qparams()
191 qparams = myobs.calculate_qparams()
256 loaded_qparams = loaded_obs.calculate_qparams()
259 self.assertEqual(myobs.calculate_qparams(), loaded_obs.calculate_qparams())
270 self.assertEqual(obs.calculate_qparams(), scripted.calculate_qparams())
276 self.assertEqual(obs.calculate_qparams(), loaded.calculate_qparams())
[all …]
Dtest_workflow_ops.py344 scale, zero_point = obs.calculate_qparams()
375 scale, zero_point = obs.calculate_qparams()
581 … self.assertEqual(loaded_fq_module.calculate_qparams(), fq_module.calculate_qparams())
671 self.assertEqual(fq_module.calculate_qparams(), scripted_module.calculate_qparams())
677 self.assertEqual(fq_module.calculate_qparams(), loaded_module.calculate_qparams())
710 scale, zero_point = obs.calculate_qparams()
870 scale, zero_point = obs.calculate_qparams()
/external/pytorch/torch/ao/quantization/
Dfake_quantize.py102 def calculate_qparams(self, **kwargs): member in FakeQuantizeBase
221 def calculate_qparams(self): member in FakeQuantize
222 return self.activation_post_process.calculate_qparams()
227 _scale, _zero_point = self.calculate_qparams()
344 def calculate_qparams(self): member in FixedQParamsFakeQuantize
395 def calculate_qparams(self) -> Tuple[torch.Tensor, torch.Tensor]: member in FusedMovingAvgObsFakeQuantize
396 return self.activation_post_process.calculate_qparams()
D_learnable_fake_quantize.py149 def calculate_qparams(self): member in _LearnableFakeQuantize
163 _scale, _zero_point = self.activation_post_process.calculate_qparams()
Dobserver.py159 def calculate_qparams(self, **kwargs): member in ObserverBase
544 def calculate_qparams(self): member in MinMaxObserver
766 def calculate_qparams(self): member in PerChannelMinMaxObserver
1310 def calculate_qparams(self): member in HistogramObserver
1423 def calculate_qparams(self): member in FixedQParamsObserver
1492 def calculate_qparams(self): member in PlaceholderObserver
1518 def calculate_qparams(self): member in RecordingObserver
1551 def calculate_qparams(self): member in NoopObserver
1578 def calculate_qparams(self): member in ReuseInputObserver
D__init__.py219 def calculate_qparams(self): member in _DerivedObserverOrFakeQuantize
/external/pytorch/benchmarks/operator_benchmark/pt/
Dqobserver_test.py113 return self.op_func.calculate_qparams()
124 return self.q_observer.calculate_qparams()
/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
Dlinear_relu.py105 …act_scale, act_zp = activation_post_process.calculate_qparams() # type: ignore[union-attr,operato…
170 …act_scale, act_zp = activation_post_process.calculate_qparams() # type: ignore[union-attr,operato…
/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
Dquantization_hooks.py76 s, z = myObserver.calculate_qparams()
170 s_ch, z_ch = myPerChannelObserver.calculate_qparams()
/external/executorch/backends/cadence/aot/quantizer/
Dutils.py130 act_scale, _ = obs_or_fqs[0].calculate_qparams()
131 weight_scale, _ = obs_or_fqs[1].calculate_qparams()
/external/pytorch/torch/ao/nn/sparse/quantized/
Dlinear.py250 …act_scale, act_zp = activation_post_process.calculate_qparams() # type: ignore[operator, union-at…
252 w_sc, w_zp = weight_post_process.calculate_qparams()
/external/pytorch/test/ao/sparsity/
Dtest_kernels.py193 qparams = qmodel.linear.qconfig.weight().calculate_qparams()
194 sqparams = sqmodel.linear.qconfig.weight().calculate_qparams()
/external/pytorch/test/onnx/
Dtest_models.py208 module.calculate_qparams()
230 module.calculate_qparams()

12