Searched refs:grad_output_buffer_data (Results 1 – 1 of 1) sorted by relevance
373 float* grad_output_buffer_data = grad_output_buffer.get(); in _vec_softmax_backward() local402 grad_output_buffer_data + dim_idx * CHUNK_SIZE; in _vec_softmax_backward()443 grad_output_buffer_data + dim_idx * CHUNK_SIZE; in _vec_softmax_backward()583 float* grad_output_buffer_data = grad_output_buffer.get(); in _vec_log_softmax_backward() local608 grad_output_buffer_data + dim_idx * CHUNK_SIZE; in _vec_log_softmax_backward()641 grad_output_buffer_data + dim_idx * CHUNK_SIZE; in _vec_log_softmax_backward()