Searched refs:max_in_row (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 93 InputT max_in_row = std::numeric_limits<InputT>::min(); in Softmax() local
95 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in Softmax()
101 static_cast<int32_t>(input_data[i * depth + c]) - max_in_row; in Softmax()
119 static_cast<int32_t>(input_data[i * depth + c]) - max_in_row; in Softmax()
150 int16_t max_in_row, int i, int c) { in SoftMaxCalculateExp() argument
151 int32_t input_diff = input_data[i * depth + c] - max_in_row; in SoftMaxCalculateExp()
178 int16_t max_in_row = std::numeric_limits<int16_t>::min(); in SoftmaxInt16() local
180 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in SoftmaxInt16()
191 SoftMaxCalculateExp(params, input_data, depth, max_in_row, i, c); in SoftmaxInt16()
|
| D | reference_ops.h | 965 uint8 max_in_row = 0; in LogSoftmax() local
967 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in LogSoftmax()
973 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in LogSoftmax()
1005 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 44 int8 max_in_row = kMinInt8; in LogSoftmax() local
46 max_in_row = in LogSoftmax()
47 std::max(max_in_row, input_data[outer_index * depth + inner_index]); in LogSoftmax()
55 max_in_row; in LogSoftmax()
83 max_in_row; in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 75 int8_t max_in_row = std::numeric_limits<int8_t>::min(); in SoftmaxHifimini() local
77 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in SoftmaxHifimini()
82 TFLITE_DCHECK(max_in_row >= input_data[i * depth + c]); in SoftmaxHifimini()
83 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini()
94 uint8_t input_diff = max_in_row - input_data[i * depth + c]; in SoftmaxHifimini()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4020 uint8 max_in_row = 0; in Softmax() local
4043 max_in_row = vget_lane_u8(max1, 0); in Softmax()
4046 max_in_row = std::max(max_in_row, input_data_ptr[c]); in Softmax()
4058 int16x8_t max_in_row_s16 = vdupq_n_s16(max_in_row); in Softmax()
4112 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
4170 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
|