Searched defs:dense_dim_opt (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 956 …ptional<c10::Layout> layout, OptionalIntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in _to_sparse_check_arguments()
1038 …nsor& self, const Tensor& self_mask, IntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_compressed()
1111 …ptional<c10::Layout> layout, OptionalIntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_with_mask()
1139 Tensor dense_to_sparse_csr(const Tensor& self, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_csr()
1146 Tensor dense_to_sparse_csc(const Tensor& self, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_csc()
1153 …se_to_sparse_bsr(const Tensor& self, IntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_bsr()
1160 …se_to_sparse_bsc(const Tensor& self, IntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse_bsc()
1167 …ptional<c10::Layout> layout, OptionalIntArrayRef blocksize, std::optional<int64_t> dense_dim_opt) { in dense_to_sparse()
1437 Tensor sparse_compressed_to_sparse_csr(const Tensor& self, std::optional<int64_t> dense_dim_opt) { in sparse_compressed_to_sparse_csr()
1450 Tensor sparse_compressed_to_sparse_csc(const Tensor& self, std::optional<int64_t> dense_dim_opt) { in sparse_compressed_to_sparse_csc()
[all …]
|