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 | 101 int32_t input_diff = in Softmax() local
103 if (input_diff >= diff_min) { in Softmax()
106 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
119 int32_t input_diff = in Softmax() local
121 if (input_diff >= diff_min) { in Softmax()
124 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
152 int32_t input_diff = input_data[i * depth + c] - max_in_row; in SoftMaxCalculateExp() local
156 input_diff, params.input_multiplier, params.input_left_shift); in SoftMaxCalculateExp()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 55 int32_t input_diff = in LogSoftmax() local
58 if (input_diff >= diff_min) { in LogSoftmax()
60 input_diff, input_multiplier, input_shift); in LogSoftmax()
83 int32_t input_diff = in LogSoftmax() local
87 if (input_diff > adjusted_diff_min) { in LogSoftmax()
89 input_diff, input_multiplier, input_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4117 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4118 if (input_diff >= diff_min) { in Softmax()
4121 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4175 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4176 if (input_diff >= diff_min) { in Softmax()
4179 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|