Searched refs:alias_into_sparse (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | SparseTensorUtils.h | 35 inline void alias_into_sparse( in alias_into_sparse() function
49 alias_into_sparse( in copy_into_sparse()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseTensor.cpp | 212 alias_into_sparse(self, indices_shallow_copy.pin_memory(), values_shallow_copy.pin_memory()); in new_with_dims_and_tensor_sparse_symint()
214 alias_into_sparse(self, indices_shallow_copy, values_shallow_copy); in new_with_dims_and_tensor_sparse_symint()
655 alias_into_sparse(dst, newIndices, newValues); in _coalesce_sparse_cpu()
|
| D | SparseTensorMath.cpp | 542 alias_into_sparse(r, r_indices, r_values); in add_out_sparse_non_contiguous()
548 alias_into_sparse(r, c._indices(), c._values()); in add_out_sparse_non_contiguous()
|
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseCUDATensorMath.cu | 426 alias_into_sparse(promoted, r_indices_, r_values_); in add_out_sparse_cuda()
435 alias_into_sparse(r_, r_indices_, r_values_); in add_out_sparse_cuda()
441 alias_into_sparse(r_, c._indices(), c._values()); in add_out_sparse_cuda()
|