Searched refs:sparse_resize_ (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
H A D | SparseUnaryOps.cpp | 116 sparse_resize_(result, input.sizes(), input.sparse_dim(), input.dense_dim()); in coalesced_unary_ufunc_out()
|
H A D | SparseTensor.cpp | 544 const SparseTensor& sparse_resize_( in sparse_resize_() function 574 sparse_resize_(self, src.sizes(), src.sparse_dim(), src.dense_dim()); in resize_as_sparse_()
|
/aosp_15_r20/external/pytorch/test/expect/ |
H A D | HasDecompTest.test_has_decomposition.expect | 1188 aten::sparse_resize_
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | sparse.rst | 1198 Tensor.sparse_resize_
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | overrides.py | 1484 Tensor.sparse_resize_: lambda self, size1, size2, dense_dim: -1,
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sparse.py | 4249 r.sparse_resize_((4, 4), 1, 1)
|