Searched defs:grad_input_vec (Results 1 – 1 of 1) sorted by relevance
335 Vec grad_input_vec = output_vec * (grad_output_vec - sum_vec); in _vec_softmax_backward() local461 Vec grad_input_vec = in _vec_softmax_backward() local544 Vec grad_input_vec = grad_output_vec - output_vec.exp() * sum_vec; in _vec_log_softmax_backward() local657 Vec grad_input_vec = in _vec_log_softmax_backward() local