Searched refs:dense_to_sparse_with_mask (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.h | 23 Tensor dense_to_sparse_with_mask(const Tensor& self, const Tensor& mask, std::optional<c10::Layout>…
|
| H A D | TensorConversions.cpp | 1111 Tensor dense_to_sparse_with_mask(const Tensor& self, const Tensor& mask, std::optional<c10::Layout>… in dense_to_sparse_with_mask() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensorMath.cpp | 362 …return at::native::dense_to_sparse_with_mask(self, dense_mask, mask.layout(), {}, mask.dense_dim()… in sparse_mask_sparse_compressed()
366 …return at::native::dense_to_sparse_with_mask(self, dense_mask, mask.layout(), blocksize, mask.dens… in sparse_mask_sparse_compressed()
|