| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax.cc | 35 int diff_min = 0; member
52 data->diff_min = -1.0 * tflite::CalculateInputRadius( in CalculateSoftmaxOpData()
99 op_params.diff_min = data->diff_min; in Softmax1DQuantized()
118 op_params.diff_min = data->diff_min; in Softmax2DQuantized()
139 op_params.diff_min = data->diff_min; in Softmax4DQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 29 const int diff_min = params.diff_min; in Softmax() local
58 if (input_diff >= diff_min) { in Softmax()
76 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/ |
| D | activations.cc | 49 int diff_min = 0; member
257 data->diff_min = -1.0 * tflite::CalculateInputRadius( in SoftmaxPrepare()
290 data->diff_min = -1.0 * tflite::CalculateInputRadius( in LogSoftmaxPrepare()
620 op_params.diff_min = data->diff_min; in Softmax1DQuantizedUint8()
637 op_params.diff_min = data->diff_min; in Softmax2DQuantizedUint8()
653 op_params.diff_min = data->diff_min; in Softmax3DQuantizedUint8()
676 op_params.diff_min = data->diff_min; in Softmax4DQuantizedUint8()
690 op_params.diff_min = data->diff_min; in Softmax1DQuantizedInt8()
704 op_params.diff_min = data->diff_min; in Softmax2DQuantizedInt8()
720 op_params.diff_min = data->diff_min; in Softmax3DQuantizedInt8()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 65 const int diff_min = params.diff_min; in Softmax() local
94 if (input_diff >= diff_min) { in Softmax()
112 if (input_diff >= diff_min) { in Softmax()
|
| D | legacy_reference_ops.h | 708 int diff_min, uint8* output_data, in Softmax() argument
713 params.diff_min = diff_min; in Softmax()
727 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
734 params.diff_min = diff_min; in LogSoftmax()
1686 int diff_min, uint8* output_data, in Softmax() argument
1689 input_beta_left_shift, diff_min, output_data, in Softmax()
1702 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1706 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
|
| D | reference_ops.h | 2607 const int diff_min = params.diff_min; in LogSoftmax() local
2636 if (input_diff >= diff_min) { in LogSoftmax()
2660 std::max(diff_min - 1, // Note use of > below instead of >= above. in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 170 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneLogSoftmaxTest() local
178 params.diff_min = diff_min; in RunOneLogSoftmaxTest()
221 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneLogSoftmaxTest() local
229 reverse_scaling_right_shift, diff_min, outer_size, inner_size, input_data, in RunOneLogSoftmaxTest()
|
| D | softmax_quantized_test.cc | 120 const int diff_min = -tflite::CalculateInputRadius(kScaledDiffIntegerBits, in RunOneSoftmaxTest() local
126 params.diff_min = diff_min; in RunOneSoftmaxTest()
|
| D | types.h | 961 int diff_min; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1444 int diff_min, uint8* output_data, in Softmax() argument
1449 params.diff_min = diff_min; in Softmax()
1454 int diff_min, uint8* output_data, in Softmax() argument
1457 input_beta_left_shift, diff_min, output_data, in Softmax()
1477 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1484 params.diff_min = diff_min; in LogSoftmax()
1491 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1495 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
|
| D | optimized_ops.h | 4456 const int diff_min = params.diff_min; in Softmax() local
4528 int32x4_t diff_min_s32 = vdupq_n_s32(diff_min); in Softmax()
4574 if (input_diff >= diff_min) { in Softmax()
4608 int16x8_t diff_min_s16 = vdupq_n_s16(diff_min); in Softmax()
4641 if (input_diff >= diff_min) { in Softmax()
4708 const int diff_min = params.diff_min; in LogSoftmax() local
4738 if (input_diff >= diff_min) { in LogSoftmax()
4762 std::max(diff_min - 1, // Note use of > below instead of >= above. in LogSoftmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4581 const int diff_min = params.diff_min; in Softmax() local
4653 int32x4_t diff_min_s32 = vdupq_n_s32(diff_min); in Softmax()
4699 if (input_diff >= diff_min) { in Softmax()
4733 int16x8_t diff_min_s16 = vdupq_n_s16(diff_min); in Softmax()
4766 if (input_diff >= diff_min) { in Softmax()
4791 int diff_min, uint8* output_data, in Softmax() argument
4796 params.diff_min = diff_min; in Softmax()
4968 const int diff_min = params.diff_min; in LogSoftmax() local
4999 if (input_diff >= diff_min) { in LogSoftmax()
5023 std::max(diff_min - 1, // Note use of > below instead of >= above. in LogSoftmax()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 916 int diff_min; member
|