Searched refs:input_diff (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | log_softmax.h | 97 int32_t input_diff = in LogSoftmax() local
99 if (input_diff >= diff_min) { in LogSoftmax()
102 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
130 int32_t input_diff = in LogSoftmax() local
132 if (input_diff > adjusted_diff_min) { in LogSoftmax()
135 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
189 int32_t input_diff = in LogSoftmaxQuantized() local
192 if (input_diff >= diff_min) { in LogSoftmaxQuantized()
194 input_diff, input_multiplier, input_left_shift); in LogSoftmaxQuantized()
218 int32_t input_diff = in LogSoftmaxQuantized() local
[all …]
|
| D | softmax.h | 100 int32_t input_diff = in Softmax() local
102 if (input_diff >= diff_min) { in Softmax()
105 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
118 int32_t input_diff = in Softmax() local
120 if (input_diff >= diff_min) { in Softmax()
123 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
151 int32_t input_diff = input_data[i * depth + c] - max_in_row; in SoftMaxCalculateExp() local
155 input_diff, params.input_multiplier, params.input_left_shift); in SoftMaxCalculateExp()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 53 int32_t input_diff = in LogSoftmax() local
56 if (input_diff >= diff_min) { in LogSoftmax()
58 input_diff, input_multiplier, input_shift); in LogSoftmax()
81 int32_t input_diff = in LogSoftmax() local
85 if (input_diff > adjusted_diff_min) { in LogSoftmax()
87 input_diff, input_multiplier, input_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4115 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4116 if (input_diff >= diff_min) { in Softmax()
4119 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4173 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4174 if (input_diff >= diff_min) { in Softmax()
4177 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|