Searched refs:indices_sparse (Results 1 – 2 of 2) sorted by relevance
119 Tensor indices_sparse(const Tensor& self) { in indices_sparse() function
980 auto indices_sparse = indices.narrow(0, d_batch_len, res_sparse_dim - d_batch_len); in intersection_binary_op_sparse_dense_out() local983 indices_sparse.view(s_indices_expanded_shape).copy_(s_indices_expanded); in intersection_binary_op_sparse_dense_out()