Searched refs:nudged_scale (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fake_quant_ops_functor.h | 87 float nudged_min, nudged_max, nudged_scale; in operator() local
89 &nudged_scale); in operator()
90 const float inv_nudged_scale = 1.0f / nudged_scale; in operator()
95 (clamped_shifted * inv_nudged_scale + 0.5f).floor() * nudged_scale + in operator()
112 float nudged_min, nudged_max, nudged_scale; in operator() local
114 &nudged_scale); in operator()
138 float nudged_min, nudged_max, nudged_scale; in operator() local
140 &nudged_scale); in operator()
141 const auto nudged_scale_repl = inputs.constant(nudged_scale); in operator()
170 float nudged_min, nudged_max, nudged_scale; in operator() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | quantization_util.cc | 322 float* nudged_scale) { in NudgeQuantizationRange() argument
326 *nudged_scale = (max - min) / (quant_max_float - quant_min_float); in NudgeQuantizationRange()
327 const float zero_point_from_min = quant_min_float - min / *nudged_scale; in NudgeQuantizationRange()
336 *nudged_min = (quant_min_float - nudged_zero_point) * (*nudged_scale); in NudgeQuantizationRange()
337 *nudged_max = (quant_max_float - nudged_zero_point) * (*nudged_scale); in NudgeQuantizationRange()
340 void FakeQuantizeArray(const float nudged_scale, const float nudged_min, in FakeQuantizeArray() argument
344 const float inv_nudged_scale = 1.0f / nudged_scale; in FakeQuantizeArray()
351 TfLiteRound(clamped_shifted * inv_nudged_scale) * nudged_scale + in FakeQuantizeArray()
|
| D | quantization_util.h | 264 float* nudged_scale);
269 void FakeQuantizeArray(const float nudged_scale, const float nudged_min,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2896 float nudged_min, nudged_max, nudged_scale; in FakeQuant() local
2898 &nudged_max, &nudged_scale); in FakeQuant()
2900 FakeQuantizeArray(nudged_scale, nudged_min, nudged_max, input_data, in FakeQuant()
|