Searched refs:input_integer_bits (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A 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 …]
|
| H A D | quantization_util.cc | 283 int input_integer_bits, in PreprocessSoftmaxScaling() argument
304 shift += (31 - input_integer_bits); in PreprocessSoftmaxScaling()
313 std::min<double>(beta * input_scale * (1 << (31 - input_integer_bits)), in PreprocessSoftmaxScaling()
322 int input_integer_bits, in PreprocessLogSoftmaxScalingExp() argument
327 PreprocessSoftmaxScaling(beta, input_scale, input_integer_bits, in PreprocessLogSoftmaxScalingExp()
338 int CalculateInputRadius(int input_integer_bits, int input_left_shift, in CalculateInputRadius() argument
341 int64_t result = (1 << input_integer_bits) - 1; in CalculateInputRadius()
342 result <<= (total_signed_bits - input_integer_bits); in CalculateInputRadius()
347 1.0 * ((1 << input_integer_bits) - 1) * in CalculateInputRadius()
348 (1LL << (total_signed_bits - input_integer_bits)) / in CalculateInputRadius()
|
| H A 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,
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 215 int32_t CalculateInputRadius(int input_integer_bits, int input_left_shift) { in CalculateInputRadius() argument
216 const double max_input_rescaled = 1.0 * ((1 << input_integer_bits) - 1) * in CalculateInputRadius()
217 (1LL << (31 - input_integer_bits)) / in CalculateInputRadius()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 134 int32_t CalculateInputRadius(int input_integer_bits, int input_left_shift);
|