Searched refs:cpu_sparse_coo_softmax_backward (Results 1 – 1 of 1) sorted by relevance
392 void cpu_sparse_coo_softmax_backward(const Tensor& grad_input, const Tensor& grad, const Tensor& ou… in cpu_sparse_coo_softmax_backward() function583 cpu_sparse_coo_softmax_backward<scalar_t, false>( in softmax_backward_sparse_cpu()603 cpu_sparse_coo_softmax_backward<scalar_t, true>( in log_softmax_backward_sparse_cpu()