Searched defs:grad_indices (Results 1 – 7 of 7) sorted by relevance
100 auto grad_indices = grad._indices(); in step() local
316 def _make_sparse(grad, grad_indices, values): argument
413 auto grad_indices = grad._indices().contiguous(); in cpu_sparse_coo_softmax_backward() local
1816 Tensor grad_indices = grad._indices(); in _sparse_sum_backward_cpu() local
467 auto grad_indices = grad._indices().contiguous(); in cuda_sparse_coo_softmax_backward() local
595 Tensor grad_indices = grad._indices(); in _sparse_sum_backward_cuda() local
1473 grad_indices, argument