Searched refs:max_in_row (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 49 int8 max_in_row = -128; in Softmax() local
51 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in Softmax()
57 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in Softmax()
75 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in Softmax()
|
| 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/kernels/internal/reference/ |
| D | softmax.h | 85 uint8 max_in_row = 0; in Softmax() local
87 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in Softmax()
93 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in Softmax()
111 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in Softmax()
|
| D | reference_ops.h | 2627 uint8 max_in_row = 0; in LogSoftmax() local
2629 max_in_row = std::max(max_in_row, input_data[i * depth + c]); in LogSoftmax()
2635 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in LogSoftmax()
2667 static_cast<int32>(input_data[i * depth + c]) - max_in_row; in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4481 uint8 max_in_row = 0; in Softmax() local
4504 max_in_row = vget_lane_u8(max1, 0); in Softmax()
4507 max_in_row = std::max(max_in_row, input_data_ptr[c]); in Softmax()
4519 int16x8_t max_in_row_s16 = vdupq_n_s16(max_in_row); in Softmax()
4573 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
4640 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
4730 uint8 max_in_row = 0; in LogSoftmax() local
4732 max_in_row = std::max(max_in_row, block_input_data[c]); in LogSoftmax()
4737 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax()
4768 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4606 uint8 max_in_row = 0; in Softmax() local
4629 max_in_row = vget_lane_u8(max1, 0); in Softmax()
4632 max_in_row = std::max(max_in_row, input_data_ptr[c]); in Softmax()
4644 int16x8_t max_in_row_s16 = vdupq_n_s16(max_in_row); in Softmax()
4698 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
4765 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax()
4991 uint8 max_in_row = 0; in LogSoftmax() local
4993 max_in_row = std::max(max_in_row, block_input_data[c]); in LogSoftmax()
4998 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax()
5029 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax()
|