Searched refs:input_integer_bits (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | log_quantized_test.cc | 56 inline int32 LogPositiveValuesViaFloat(int32 input_val, int input_integer_bits, in LogPositiveValuesViaFloat() argument
59 static_cast<double>(input_val) * 0.5 / (1 << (30 - input_integer_bits))); in LogPositiveValuesViaFloat()
73 const string& check_label, int input_integer_bits, in CheckOutputData() argument
94 << static_cast<double>(test_input[i]) / (1 << (31 - input_integer_bits)) in CheckOutputData()
153 void RunSingleTest(const std::vector<int32>& test_input, int input_integer_bits, in RunSingleTest() argument
159 switch (input_integer_bits) { in RunSingleTest()
191 ASSERT_LE(input_integer_bits, 30) in RunSingleTest()
192 << "Input integer bits not handled: " << input_integer_bits; in RunSingleTest()
197 void RunSingleTest(const std::vector<int32>& test_input, int input_integer_bits, in RunSingleTest() argument
202 return RunSingleTest<K>(test_input, input_integer_bits, check_label, \ in RunSingleTest()
[all …]
|
| D | quantization_util.cc | 255 int input_integer_bits, in PreprocessSoftmaxScaling() argument
270 shift += (31 - input_integer_bits); in PreprocessSoftmaxScaling()
278 beta * input_scale * (1 << (31 - input_integer_bits)), (1ll << 31) - 1.0); in PreprocessSoftmaxScaling()
286 int input_integer_bits, in PreprocessLogSoftmaxScalingExp() argument
291 PreprocessSoftmaxScaling(beta, input_scale, input_integer_bits, in PreprocessLogSoftmaxScalingExp()
302 int CalculateInputRadius(int input_integer_bits, int input_left_shift) { in CalculateInputRadius() argument
304 int64_t result = (1 << input_integer_bits) - 1; in CalculateInputRadius()
305 result <<= (31 - input_integer_bits); in CalculateInputRadius()
309 const double max_input_rescaled = 1.0 * ((1 << input_integer_bits) - 1) * in CalculateInputRadius()
310 (1ll << (31 - input_integer_bits)) / in CalculateInputRadius()
|
| D | quantization_util.h | 241 int input_integer_bits,
245 int input_integer_bits,
255 int CalculateInputRadius(int input_integer_bits, int input_left_shift);
|