| /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
63 *nudged_max = (quant_max_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()
96 auto clamped = inputs.cwiseMin(nudged_max).cwiseMax(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()
142 float nudged_min, nudged_max, nudged_scale, inv_nudged_scale; in operator() local
143 Nudge(min_val, max_val, quant_min, quant_max, &nudged_min, &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
375 *nudged_max = (quant_max_float - nudged_zero_point) * (*nudged_scale); in NudgeQuantizationRange()
379 const float nudged_max, const float* input_data, in FakeQuantizeArray() argument
386 const float clamped = std::min(nudged_max, std::max(nudged_min, src_val)); in FakeQuantizeArray()
|
| D | quantization_util.h | 264 float* nudged_min, float* nudged_max,
271 const float nudged_max, const float* input_data,
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_fake_quant.cc | 131 const float nudged_max = (quantized_max - 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
45 *nudged_max = (quant_max - nudged_zero_point) * (*scale); in CpuNudge()
52 xla::XlaOp* nudged_min, xla::XlaOp* nudged_max, in XlaNudge() argument
67 *nudged_max = xla::Mul(xla::Sub(quant_max, 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
557 &nudged_max, &nudged_scale); 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
101 *nudged_max = (quant_max_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()
3272 Value cst_max = getTosaConstTensorSingleF32(rewriter, op, nudged_max); in convertFakeQuantOp()
|