Searched refs:input_diff (Results 1 – 5 of 5) sorted by relevance
| /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/reference/ |
| 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()
|
| D | reference_ops.h | 972 int32 input_diff = in LogSoftmax() local
974 if (input_diff >= diff_min) { in LogSoftmax()
977 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
1004 int32 input_diff = in LogSoftmax() local
1006 if (input_diff > adjusted_diff_min) { in LogSoftmax()
1009 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 83 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini() local
86 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
94 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini() local
97 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4112 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4113 if (input_diff >= diff_min) { in Softmax()
4116 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4170 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4171 if (input_diff >= diff_min) { in Softmax()
4174 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|