Searched full:sparsecsrmeta (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 355 {"SparseCsrMeta", c10::DispatchKey::SparseCsrMeta}, in parseDispatchKey()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorImpl.cpp | 59 || (key_set.has(DispatchKey::SparseCsrMeta) && device().type() == kMeta) in SparseCsrTensorImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 553 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: add_sparse_csr
565 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: add_sparse_csr_
580 SparseCsrCPU, SparseCsrMeta: add_out_sparse_compressed_cpu
2374 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: empty_sparse_compressed
2480 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: empty_like_sparse_csr
5724 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: sum_csr
6818 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: zero_sparse_csr_
7217 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: sparse_dim_sparse_csr
7234 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: dense_dim_sparse_csr
7251 SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: _nnz_sparse_csr
[all …]
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 191 SparseCsrMeta = auto() variable in DispatchKey
296 DispatchKey.SparseCsrMeta,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensor.cpp | 364 dispatch_key = DispatchKey::SparseCsrMeta; in new_compressed_tensor()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 13395 …dError: Could not run 'aten::sparse_sampled_addmm' with arguments from the 'SparseCsrMeta' backend.
13442 …Error: Could not run 'aten::_sparse_mm_reduce_impl' with arguments from the 'SparseCsrMeta' backend
|