Home
last modified time | relevance | path

Searched defs: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
130 int32_t input_diff = in LogSoftmax() local
189 int32_t input_diff = in LogSoftmaxQuantized() local
218 int32_t input_diff = in LogSoftmaxQuantized() local
H A Dsoftmax.h101 int32_t input_diff = in Softmax() local
119 int32_t input_diff = in Softmax() local
152 int32_t input_diff = input_data[i * depth + c] - max_in_row; in SoftMaxCalculateExp() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dlog_softmax.h55 int32_t input_diff = in LogSoftmax() local
83 int32_t input_diff = in LogSoftmax() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSoftmax.cpp146 int32_t input_diff = static_cast<int32_t>(*p) - maxValue; in softmaxQuant8Impl() local
174 int32_t input_diff = static_cast<int32_t>(*p) - maxValue; in softmaxQuant8Impl() local
/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
4175 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local