Searched refs:output_vec_o2 (Results 1 – 1 of 1) sorted by relevance
739 … Vec output_vec_o2 = Vec::loadu(temp_vec_input_data + d*vectorized_step + Vec().size()); in _vec_softmax() local741 output_vec_o2 = (output_vec_o2 - max_vec_o2).exp(); in _vec_softmax()743 output_vec_o2.store(temp_vec_output_data + d*vectorized_step + Vec().size()); in _vec_softmax()746 sum_vec_o2 = sum_vec_o2 + output_vec_o2; in _vec_softmax()751 … Vec output_vec_o2 = Vec::loadu(temp_vec_output_data + d*vectorized_step + Vec().size()); in _vec_softmax() local753 output_vec_o2 = output_vec_o2/sum_vec_o2; in _vec_softmax()754 … Vec16 output_vec_bf16 = vec::convert_from_float<scalar_t>(output_vec_o1, output_vec_o2); in _vec_softmax()