Home
last modified time | relevance | path

Searched refs:temp_vec_output_data (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp709 float* temp_vec_output_data = temp_vec_output.get(); in _vec_softmax() local
742 output_vec_o1.store(temp_vec_output_data + d*vectorized_step); in _vec_softmax()
743 output_vec_o2.store(temp_vec_output_data + d*vectorized_step + Vec().size()); in _vec_softmax()
750 Vec output_vec_o1 = Vec::loadu(temp_vec_output_data + d*vectorized_step); in _vec_softmax()
751 … Vec output_vec_o2 = Vec::loadu(temp_vec_output_data + d*vectorized_step + Vec().size()); in _vec_softmax()