| /external/tensorflow/tensorflow/core/kernels/ |
| D | fake_quant_ops_functor.h | 43 float* nudged_min, float* nudged_max, float* scale, float* inv_scale) { in Nudge() argument
62 *nudged_min = (quant_min_float - nudged_zero_point) * (*scale); in Nudge()
90 float nudged_min, nudged_max, nudged_scale, inv_nudged_scale; in operator() local
91 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
94 const float quant_zero = floor(-nudged_min * inv_nudged_scale + 0.5f); in operator()
96 auto clamped = inputs.cwiseMin(nudged_max).cwiseMax(nudged_min); in operator()
97 auto clamped_shifted = clamped - nudged_min; in operator()
116 float nudged_min, nudged_max, nudged_scale, inv_nudged_scale; in operator() local
117 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
121 (inputs >= nudged_min && inputs <= nudged_max) in operator()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | quantization_util.cc | 359 float* nudged_min, float* nudged_max, in NudgeQuantizationRange() argument
374 *nudged_min = (quant_min_float - nudged_zero_point) * (*nudged_scale); in NudgeQuantizationRange()
378 void FakeQuantizeArray(const float nudged_scale, const float nudged_min, in FakeQuantizeArray() argument
386 const float clamped = std::min(nudged_max, std::max(nudged_min, src_val)); in FakeQuantizeArray()
387 const float clamped_shifted = clamped - nudged_min; in FakeQuantizeArray()
390 nudged_min; in FakeQuantizeArray()
|
| D | quantization_util.h | 264 float* nudged_min, float* nudged_max,
270 void FakeQuantizeArray(const float nudged_scale, const float nudged_min,
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_fake_quant.cc | 130 const float nudged_min = (quantized_min - qparams.zero_point) * scale; in Run() local
132 tflite::FakeQuantizeArray(scale, nudged_min, nudged_max, in Run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 30 const float quant_max, float* nudged_min, float* nudged_max, in CpuNudge() argument
44 *nudged_min = (quant_min - nudged_zero_point) * (*scale); in CpuNudge()
52 xla::XlaOp* nudged_min, xla::XlaOp* nudged_max, in XlaNudge() argument
66 *nudged_min = xla::Mul(xla::Sub(quant_min, nudged_zero_point), *scale); in XlaNudge()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 555 float nudged_min, nudged_max, nudged_scale; in FakeQuant() local
556 NudgeQuantizationRange(rmin, rmax, quant_min, quant_max, &nudged_min, in FakeQuant()
559 FakeQuantizeArray(nudged_scale, nudged_min, nudged_max, input_data, in FakeQuant()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 83 const int quant_max, float* nudged_min, float* nudged_max, in tensorflow_nudge() argument
100 *nudged_min = (quant_min_float - nudged_zero_point) * (*scale); in tensorflow_nudge()
3267 float nudged_min, nudged_max, nudged_scale; in convertFakeQuantOp() local
3268 tensorflow_nudge(min, max, qmin, qmax, &nudged_min, &nudged_max, in convertFakeQuantOp()
3271 Value cst_min = getTosaConstTensorSingleF32(rewriter, op, nudged_min); in convertFakeQuantOp()
|