Searched full:sparsemeta (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 347 {"SparseMeta", c10::DispatchKey::SparseMeta}, in parseDispatchKey()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 552 SparseCPU, SparseCUDA, SparseMeta: add_sparse
564 SparseCPU, SparseCUDA, SparseMeta: add_sparse_
578 SparseCPU, SparseMeta: add_out_sparse_cpu
2373 SparseCPU, SparseCUDA, SparseMeta: empty_sparse
2479 SparseCPU, SparseCUDA, SparseMeta: empty_like_sparse_coo
5723 SparseCPU, SparseCUDA, SparseMeta: sum_coo
6515 SparseCPU, SparseCUDA, SparseMeta: zeros_sparse_out
6817 SparseCPU, SparseCUDA, SparseMeta: zero_sparse_
7161 SparseCPU, SparseCUDA, SparseMeta, Meta: new_with_dims_sparse
7166 SparseCPU, SparseCUDA, SparseMeta, Meta: new_with_dims_and_tensor_sparse_symint
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseUnaryOps.cpp | 208 TORCH_CHECK_NOT_IMPLEMENTED(0, "nyi isinf for SparseMeta"); in isinf_sparse_meta()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | EmptyTensor.cpp | 363 // NB: because there is no SparseMeta (yet), non-strided layout is in empty_meta()
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 176 SparseMeta = auto() variable in DispatchKey
295 DispatchKey.SparseMeta,
|