Home
last modified time | relevance | path

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

/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/reference/
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
Dreference_ops.h972 int32 input_diff = in LogSoftmax() local
1004 int32 input_diff = in LogSoftmax() local
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc83 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini() local
94 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4112 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4170 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local