Home
last modified time | relevance | path

Searched defs:input_diff (Results 1 – 6 of 6) 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
Dsoftmax.h56 int32 input_diff = in Softmax() local
74 int32 input_diff = in Softmax() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h92 int32 input_diff = in Softmax() local
110 int32 input_diff = in Softmax() local
Dreference_ops.h2634 int32 input_diff = in LogSoftmax() local
2666 int32 input_diff = in LogSoftmax() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h4573 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4640 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4737 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
4768 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4698 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4765 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4998 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
5029 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local