Home
last modified time | relevance | path

Searched defs:input_diff (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
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
Dsoftmax.h100 int32_t input_diff = in Softmax() local
118 int32_t input_diff = in Softmax() local
151 int32_t input_diff = input_data[i * depth + c] - max_in_row; in SoftMaxCalculateExp() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dlog_softmax.h53 int32_t input_diff = in LogSoftmax() local
81 int32_t input_diff = in LogSoftmax() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4115 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4173 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local