Searched refs:outer_index (Results 1 – 2 of 2) sorted by relevance
43 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in LogSoftmax() local47 std::max(max_in_row, input_data[outer_index * depth + inner_index]); in LogSoftmax()54 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmax()82 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmax()99 output_data[outer_index * depth + inner_index] = in LogSoftmax()102 output_data[outer_index * depth + inner_index] = kMinInt8; in LogSoftmax()
32 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in L2Normalization() local39 input_data[depth * outer_index + inner_index] - input_zero_point; in L2Normalization()49 input_data[depth * outer_index + inner_index] - input_zero_point; in L2Normalization()57 output_data[depth * outer_index + inner_index] = in L2Normalization()