Searched refs:sparse_resize_and_clear_ (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorFactories.cpp | 339 result.sparse_resize_and_clear_(size, size.size(), 0); in empty_out()
1348 result.sparse_resize_and_clear_(size, size.size(), 0.); in zeros_sparse_out()
1357 result.sparse_resize_and_clear_(size, size.size(), 0.); in zeros_out()
1384 res.sparse_resize_and_clear_( in zeros_like()
1387 res.sparse_resize_and_clear_( in zeros_like()
1390 res.sparse_resize_and_clear_(self.sizes(), self.sizes().size(), 0); in zeros_like()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseMatMul.cpp | 282 output.sparse_resize_and_clear_({mat1_.size(0), mat2_.size(1)}, mat1_.sparse_dim(), 0); in sparse_sparse_matmul_cpu()
|
| H A D | SparseTensor.cpp | 553 const SparseTensor& sparse_resize_and_clear_( in sparse_resize_and_clear_() function
866 result.sparse_resize_and_clear_( in empty_like_sparse_coo()
|
| H A D | SparseTensorMath.cpp | 88 self.sparse_resize_and_clear_(self.sizes(), self.sparse_dim(), self.dense_dim()); in zero_sparse_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseMatMul.cu | 811 output.sparse_resize_and_clear_({mat1_.size(0), mat2_.size(1)}, mat1_.sparse_dim(), 0); in sparse_sparse_matmul_cuda()
|
| /aosp_15_r20/external/pytorch/ |
| H A D | pt_ops.bzl | 656 "aten::sparse_resize_and_clear_",
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1191 aten::sparse_resize_and_clear_
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 4787 res.sparse_resize_and_clear_(
4791 res.sparse_resize_and_clear_(self.size(), self.dim(), 0)
|
| H A D | overrides.py | 1485 Tensor.sparse_resize_and_clear_: lambda self, size1, size2, dense_dim: -1,
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | sparse.rst | 1199 Tensor.sparse_resize_and_clear_
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 4250 r.sparse_resize_and_clear_((4, 4, 4), 2, 1)
|