Searched refs:_to_sparse_check_arguments (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 929 void _to_sparse_check_arguments(const std::string& funcname, const Tensor& self, const int64_t spar… in _to_sparse_check_arguments() function
956 void _to_sparse_check_arguments(const std::string& funcname, const Tensor& self, std::optional<c10:… in _to_sparse_check_arguments() function
1118 _to_sparse_check_arguments("dense_to_sparse_with_mask", self, layout, blocksize, dense_dim_opt); in dense_to_sparse_with_mask()
1141 _to_sparse_check_arguments("dense_to_sparse_csr", self, layout_to, {}, dense_dim_opt); in dense_to_sparse_csr()
1148 _to_sparse_check_arguments("dense_to_sparse_csc", self, layout_to, {}, dense_dim_opt); in dense_to_sparse_csc()
1155 _to_sparse_check_arguments("dense_to_sparse_bsr", self, layout_to, blocksize, dense_dim_opt); in dense_to_sparse_bsr()
1162 _to_sparse_check_arguments("dense_to_sparse_bsc", self, layout_to, blocksize, dense_dim_opt); in dense_to_sparse_bsc()
1170 _to_sparse_check_arguments("dense_to_sparse", self, layout, blocksize, dense_dim_opt); in dense_to_sparse()
1192 _to_sparse_check_arguments("dense_to_sparse", self, sparse_dim); in dense_to_sparse()
1440 _to_sparse_check_arguments("sparse_compressed_to_sparse_csr", self, layout_to, {}, dense_dim_opt); in sparse_compressed_to_sparse_csr()
[all …]
|