Searched refs:quant_max (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | fake_quant_ops_functor.h | 42 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 …]
|
D | fake_quant_ops.cc | 153 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/ |
D | fake_quantize_ops.cc | 28 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()
|