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()
154 void RunSingleTest(const std::vector<int32>& test_input, int input_integer_bits, in RunSingleTest() argument
160 switch (input_integer_bits) { in RunSingleTest()
192 ASSERT_LE(input_integer_bits, 30) in RunSingleTest()
193 << "Input integer bits not handled: " << input_integer_bits; in RunSingleTest()
198 void RunSingleTest(const std::vector<int32>& test_input, int input_integer_bits, in RunSingleTest() argument
203 return RunSingleTest<K>(test_input, input_integer_bits, check_label, \ in RunSingleTest()
[all …]
|
| D | quantization_util.cc | 270 int input_integer_bits, in PreprocessSoftmaxScaling() argument
285 shift += (31 - input_integer_bits); in PreprocessSoftmaxScaling()
293 beta * input_scale * (1 << (31 - input_integer_bits)), (1ll << 31) - 1.0); in PreprocessSoftmaxScaling()
301 int input_integer_bits, in PreprocessLogSoftmaxScalingExp() argument
306 PreprocessSoftmaxScaling(beta, input_scale, input_integer_bits, in PreprocessLogSoftmaxScalingExp()
317 int CalculateInputRadius(int input_integer_bits, int input_left_shift, in CalculateInputRadius() argument
320 int64_t result = (1 << input_integer_bits) - 1; in CalculateInputRadius()
321 result <<= (total_signed_bits - input_integer_bits); in CalculateInputRadius()
326 1.0 * ((1 << input_integer_bits) - 1) * in CalculateInputRadius()
327 (1ll << (total_signed_bits - 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,
|