| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| 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 | 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 | softmax_quantized_test.cc | 125 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneSoftmaxTest() local
132 params.diff_min = diff_min; in RunOneSoftmaxTest()
|
| D | types.h | 1088 int diff_min; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 72 const int diff_min = params.diff_min; in Softmax() local
102 if (input_diff >= diff_min) { in Softmax()
120 if (input_diff >= diff_min) { in Softmax()
|
| D | legacy_reference_ops.h | 758 int diff_min, uint8* output_data, in Softmax() argument
763 params.diff_min = diff_min; in Softmax()
777 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
784 params.diff_min = diff_min; in LogSoftmax()
1777 int diff_min, uint8* output_data, in Softmax() argument
1780 input_beta_left_shift, diff_min, output_data, in Softmax()
1793 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1797 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
|
| D | reference_ops.h | 945 const int diff_min = params.diff_min; in LogSoftmax() local
974 if (input_diff >= diff_min) { in LogSoftmax()
998 std::max(diff_min - 1, // Note use of > below instead of >= above. in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 57 op_data.diff_min, in SoftmaxQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax_common.cc | 77 op_data->diff_min = in CalculateSoftmaxParams()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3995 const int diff_min = params.diff_min; in Softmax() local
4067 int32x4_t diff_min_s32 = vdupq_n_s32(diff_min); in Softmax()
4113 if (input_diff >= diff_min) { in Softmax()
4138 int16x8_t diff_min_s16 = vdupq_n_s16(diff_min); in Softmax()
4171 if (input_diff >= diff_min) { in Softmax()
4209 int diff_min, uint8* output_data, in Softmax() argument
4214 params.diff_min = diff_min; in Softmax()
4219 int diff_min, uint8* output_data, in Softmax() argument
4222 input_beta_left_shift, diff_min, output_data, in Softmax()
4242 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 230 op_data->params.diff_min, op_data->params.input_left_shift, in EvalHifi4()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 96 int diff_min; member
1187 int diff_min;
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 67 int diff_min = 0; member
1234 data->diff_min, outer_size, depth, GetTensorData<int8_t>(input), in LogSoftmaxEval()
|