| /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()
52 quant_min_float, quant_max, in Nudge()
58 return static_cast<uint16>(quant_max); in Nudge()
84 const float max, const int quant_min, const int quant_max, in operator()
91 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
110 const int quant_min, const int quant_max, in operator()
117 Nudge(min, max, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
133 const int quant_min, const int quant_max, in operator()
143 Nudge(min_val, max_val, quant_min, quant_max, &nudged_min, &nudged_max, in operator()
[all …]
|
| D | fake_quant_ops.cc | 166 const float min, const float max, const int quant_min, const int quant_max,
176 const int quant_min, const int quant_max,
309 const int quant_max, typename TTypes<float>::Flat output);
323 const int quant_max, typename TTypes<float>::Flat backprops_wrt_input,
477 const int quant_max, typename TTypes<float>::Matrix outputs);
491 const int quant_min, const int quant_max,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | quantize_and_dequantize_test.cc | 42 const int quant_max = (1 << num_bits) - 1; in TEST() local
45 quant_min, quant_max, &attr.min, &attr.max, in TEST()
73 const int quant_max = (1 << num_bits) - 1; in TEST() local
76 quant_min, quant_max, &attr.min, &attr.max, in TEST()
104 const int quant_max = (1 << num_bits) - 1; in TEST() local
107 quant_min, quant_max, &attr.min, &attr.max, in TEST()
135 const int quant_max = (1 << num_bits) - 1; in TEST() local
138 quant_min, quant_max, &attr.min, &attr.max, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | quantize_and_dequantize_test_util.cc | 37 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim2Bits8Test() local
40 quant_min, quant_max, &attr.min, &attr.max, in QuantAndDequant_Dim2Bits8Test()
74 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
77 quant_min, quant_max, &attr.min, &attr.max, in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
110 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim3Bits16Test() local
113 quant_min, quant_max, &attr.min, &attr.max, in QuantAndDequant_Dim3Bits16Test()
147 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim2Bits16_NegativeRangeTest() local
150 quant_min, quant_max, &attr.min, &attr.max, in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
|
| /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
32 *scale = (max - min) / (quant_max - quant_min); in CpuNudge()
38 } else if (zero_point_from_min >= quant_max) { in CpuNudge()
39 nudged_zero_point = quant_max; in CpuNudge()
45 *nudged_max = (quant_max - nudged_zero_point) * (*scale); in CpuNudge()
60 xla::XlaOp quant_max = in XlaNudge() local
64 xla::Select(xla::Ge(zero_point_from_min, quant_max), in XlaNudge()
65 quant_max, xla::Round(zero_point_from_min))); in XlaNudge()
67 *nudged_max = xla::Mul(xla::Sub(quant_max, nudged_zero_point), *scale); in XlaNudge()
101 const float quant_max = (1 << num_bits) - 1; in FakeQuantWithMinMaxArgsGradOp() local
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils_test.cc | 83 const int quant_max = 127; in TEST_F() local
85 GetAsymmetricQuantizationParams(float_min, float_max, quant_min, quant_max, in TEST_F()
107 const int quant_max = 127; in TEST_F() local
109 GetAsymmetricQuantizationParams(float_min, float_max, quant_min, quant_max, in TEST_F()
130 const int quant_max = 127; in TEST_F() local
132 GetAsymmetricQuantizationParams(float_min, float_max, quant_min, quant_max, in TEST_F()
151 const int quant_max = 127; in TEST_F() local
153 GetAsymmetricQuantizationParams(float_min, float_max, quant_min, quant_max, in TEST_F()
166 EXPECT_LT(zero_point, quant_max); in TEST_F()
173 const int quant_max = 127; in TEST_F() local
[all …]
|
| D | quantization_utils.h | 37 float min, float max, const int quant_min, const int quant_max,
|
| D | quantization_utils.cc | 66 float min, float max, const int quant_min, const int quant_max, in GetAsymmetricQuantizationParams() argument
69 const float quant_max_float = static_cast<float>(quant_max); in GetAsymmetricQuantizationParams()
83 zero_point = static_cast<int64_t>(quant_max); in GetAsymmetricQuantizationParams()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | quantization_util.cc | 358 const int quant_min, const int quant_max, in NudgeQuantizationRange() argument
363 const float quant_max_float = static_cast<float>(quant_max); in NudgeQuantizationRange()
370 nudged_zero_point = static_cast<uint16_t>(quant_max); in NudgeQuantizationRange()
|
| D | quantization_util.h | 263 const int quant_min, const int quant_max,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 394 auto quant_max = rewriter.create<ConstOp>( in matchAndRewrite() local
420 op.getLoc(), scalar_ty, mid_rounded, quant_min, quant_max); in matchAndRewrite()
425 rewriter.create<SubOp>(op.getLoc(), quant_max, nudged_zero_point_val); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 554 int quant_max = (1 << num_bits) - 1; in FakeQuant() local
556 NudgeQuantizationRange(rmin, rmax, quant_min, quant_max, &nudged_min, 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
86 const float quant_max_float = static_cast<float>(quant_max); in tensorflow_nudge()
90 quant_min_float, quant_max, in tensorflow_nudge()
96 return static_cast<uint16_t>(quant_max); in tensorflow_nudge()
|