Home
last modified time | relevance | path

Searched refs:input_diff (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlog_softmax.h97 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 …]
H A Dsoftmax.h101 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dlog_softmax.h55 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()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSoftmax.cpp146 int32_t input_diff = static_cast<int32_t>(*p) - maxValue; in softmaxQuant8Impl() local
147 if (input_diff >= diffMin) { in softmaxQuant8Impl()
150 input_diff, inputMultiplier, inputLeftShift); in softmaxQuant8Impl()
174 int32_t input_diff = static_cast<int32_t>(*p) - maxValue; in softmaxQuant8Impl() local
175 if (input_diff >= diffMin) { in softmaxQuant8Impl()
178 input_diff, inputMultiplier, inputLeftShift); in softmaxQuant8Impl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h4117 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()