Searched defs:r_indices (Results 1 – 1 of 1) sorted by relevance
445 Tensor r_indices = at::empty({src.sparse_dim(), max_nnz}, t._indices().options()); in add_out_sparse_contiguous() local540 Tensor r_indices = at::cat({t._indices(), src._indices()}, 1); in add_out_sparse_non_contiguous() local1108 Tensor r_indices = at::empty({sparse_dim, max_nnz}, t_indices.options()); in mul_out_sparse_cpu() local