Searched refs:grad_output_buffer_ptr (Results 1 – 1 of 1) sorted by relevance
401 float* grad_output_buffer_ptr = in _vec_softmax_backward() local422 grad_output_fvec0.store(grad_output_buffer_ptr + d1); in _vec_softmax_backward()424 grad_output_buffer_ptr + d1 + fVec::size()); in _vec_softmax_backward()432 grad_output_buffer_ptr[d1] = grad_output_val; in _vec_softmax_backward()442 float* grad_output_buffer_ptr = in _vec_softmax_backward() local451 fVec grad_output_fvec0 = fVec::loadu(grad_output_buffer_ptr + d2); in _vec_softmax_backward()453 fVec::loadu(grad_output_buffer_ptr + d2 + fVec::size()); in _vec_softmax_backward()466 … grad_input_ptr[d2] = output_buffer_ptr[d2] * (grad_output_buffer_ptr[d2] - tmp_sum_data[d2]); in _vec_softmax_backward()607 float* grad_output_buffer_ptr = in _vec_log_softmax_backward() local623 grad_output_fvec0.store(grad_output_buffer_ptr + d1); in _vec_log_softmax_backward()[all …]