Home
last modified time | relevance | path

Searched refs:quant_max (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops_functor.h42 const float min, const float max, const int quant_min, const int quant_max, in Nudge() argument
45 const float quant_max_float = static_cast<float>(quant_max); in Nudge()
49 quant_min_float, quant_max, in Nudge()
55 return static_cast<uint16>(quant_max); in Nudge()
81 const float max, const int quant_min, const int quant_max, in operator()
88 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
106 const int quant_min, const int quant_max, in operator()
113 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
129 const int quant_min, const int quant_max, in operator()
139 Nudge(min_val, max_val, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
[all …]
Dfake_quant_ops.cc153 const float min, const float max, const int quant_min, const int quant_max,
163 const int quant_min, const int quant_max,
274 const int quant_max, typename TTypes<float>::Flat output);
288 const int quant_max, typename TTypes<float>::Flat backprops_wrt_input,
420 const int quant_max, typename TTypes<float>::Matrix outputs);
434 const int quant_min, const int quant_max,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc28 const float quant_max, float* nudged_min, float* nudged_max, in CpuNudge() argument
30 *scale = (max - min) / (quant_max - quant_min); in CpuNudge()
36 } else if (zero_point_from_min >= quant_max) { in CpuNudge()
37 nudged_zero_point = quant_max; in CpuNudge()
43 *nudged_max = (quant_max - nudged_zero_point) * (*scale); in CpuNudge()
61 xla::ComputationDataHandle quant_max = in XlaNudge() local
65 b->Select(b->Ge(zero_point_from_min, quant_max), quant_max, in XlaNudge()
68 *nudged_max = b->Mul(b->Sub(quant_max, nudged_zero_point), *scale); in XlaNudge()
152 const float quant_max = (1 << num_bits) - 1; in FakeQuantWithMinMaxArgsGradOp() local
157 CpuNudge(input_min, input_max, quant_min, quant_max, &nudged_input_min_, in FakeQuantWithMinMaxArgsGradOp()