Home
last modified time | relevance | path

Searched refs:sparse_resize_ (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseUnaryOps.cpp116 sparse_resize_(result, input.sizes(), input.sparse_dim(), input.dense_dim()); in coalesced_unary_ufunc_out()
H A DSparseTensor.cpp544 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 DHasDecompTest.test_has_decomposition.expect1188 aten::sparse_resize_
/aosp_15_r20/external/pytorch/docs/source/
H A Dsparse.rst1198 Tensor.sparse_resize_
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py1484 Tensor.sparse_resize_: lambda self, size1, size2, dense_dim: -1,
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py4249 r.sparse_resize_((4, 4), 1, 1)