| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/ |
| D | softmax.cc | 41 const int diff_min = params.diff_min; in Softmax() local
71 if (input_diff >= diff_min) { in Softmax()
89 if (input_diff >= diff_min) { in Softmax()
125 int diff_min = 0; member
156 data->diff_min = -1.0 * tflite::CalculateInputRadius( in CalculateSoftmaxOpData()
173 op_params.diff_min = data->diff_min; in Softmax2DQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax.cc | 37 int diff_min = 0; member
65 data->diff_min = -1.0 * tflite::CalculateInputRadius( in CalculateSoftmaxOpData()
112 op_params.diff_min = data->diff_min; in Softmax1DQuantized()
143 op_params.diff_min = data->diff_min; in Softmax2DQuantized()
176 op_params.diff_min = data->diff_min; in Softmax4DQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 30 const int diff_min = params.diff_min; in Softmax() local
60 if (input_diff >= diff_min) { in Softmax()
78 if (input_diff >= diff_min) { in Softmax()
|
| D | log_softmax.h | 25 int32_t diff_min, int32_t outer_size, int32_t depth, in LogSoftmax() argument
56 if (input_diff >= diff_min) { in LogSoftmax()
76 diff_min - 1, in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | softmax_quantized_test.cc | 123 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneSoftmaxTest() local
130 params.diff_min = diff_min; in RunOneSoftmaxTest()
198 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneSoftmaxTestInt8() local
204 params.diff_min = diff_min; in RunOneSoftmaxTestInt8()
|
| D | logsoftmax_quantized_test.cc | 170 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneLogSoftmaxTest() local
179 params.diff_min = diff_min; in RunOneLogSoftmaxTest()
227 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneLogSoftmaxTest() local
235 reverse_scaling_right_shift, diff_min, outer_size, inner_size, input_data, in RunOneLogSoftmaxTest()
|
| D | types.h | 1023 int diff_min; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | softmax.h | 32 const int diff_min = params.diff_min; in Softmax() local
104 int32x4_t diff_min_s32 = vdupq_n_s32(diff_min); in Softmax()
149 if (input_diff >= diff_min) { in Softmax()
174 int16x8_t diff_min_s16 = vdupq_n_s16(diff_min); in Softmax()
211 if (input_diff >= diff_min) { in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 69 const int diff_min = params.diff_min; in Softmax() local
98 if (input_diff >= diff_min) { in Softmax()
116 if (input_diff >= diff_min) { in Softmax()
|
| D | legacy_reference_ops.h | 745 int diff_min, uint8* output_data, in Softmax() argument
750 params.diff_min = diff_min; in Softmax()
764 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
771 params.diff_min = diff_min; in LogSoftmax()
1807 int diff_min, uint8* output_data, in Softmax() argument
1810 input_beta_left_shift, diff_min, output_data, in Softmax()
1823 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1827 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
|
| D | reference_ops.h | 1626 const int diff_min = params.diff_min; in LogSoftmax() local
1655 if (input_diff >= diff_min) { in LogSoftmax()
1679 std::max(diff_min - 1, // Note use of > below instead of >= above. in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3982 const int diff_min = params.diff_min; in Softmax() local
4054 int32x4_t diff_min_s32 = vdupq_n_s32(diff_min); in Softmax()
4100 if (input_diff >= diff_min) { in Softmax()
4125 int16x8_t diff_min_s16 = vdupq_n_s16(diff_min); in Softmax()
4158 if (input_diff >= diff_min) { in Softmax()
4196 int diff_min, uint8* output_data, in Softmax() argument
4201 params.diff_min = diff_min; in Softmax()
4206 int diff_min, uint8* output_data, in Softmax() argument
4209 input_beta_left_shift, diff_min, output_data, in Softmax()
4229 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 62 int diff_min = 0; member
598 data->diff_min = in LogSoftmaxPrepare()
1037 data->diff_min, outer_size, depth, GetTensorData<int8_t>(input), in LogSoftmaxEval()
|