| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 26 * or "SparseCUDA"; backend in torch.backends is something like "MKL" or
38 SparseCUDA, enumerator
95 } else if (t == DispatchKey::SparseCUDA) { in dispatchKeyToBackend()
96 return Backend::SparseCUDA; in dispatchKeyToBackend()
173 case Backend::SparseCUDA: in backendToDispatchKey()
174 return DispatchKey::SparseCUDA; in backendToDispatchKey()
229 case Backend::SparseCUDA: in backendToDeviceType()
310 case Backend::SparseCUDA: in toString()
311 return "SparseCUDA"; in toString()
363 case Backend::SparseCUDA: in isSparse()
|
| H A D | DispatchKey.cpp | 343 {"SparseCUDA", c10::DispatchKey::SparseCUDA}, in parseDispatchKey()
|
| H A D | Layout.h | 34 case Backend::SparseCUDA: in layout_from_backend()
|
| H A D | DispatchKey.h | 544 // DispatchKeySet, while `SparseCPU`, `SparseCUDA`, etc all map to individual
|
| H A D | DispatchKeySet.h | 81 // - "Sparse": SparseCPU, SparseCUDA, ...
|
| H A D | TensorImpl.h | 2027 * compatible with SparseCUDA.
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 340 SparseCPU, SparseCUDA: abs_sparse
350 SparseCPU, SparseCUDA: abs_sparse_
359 SparseCPU, SparseCUDA: abs_sparse_out
427 SparseCPU, SparseCUDA: sgn_sparse
436 SparseCPU, SparseCUDA: sgn_sparse_
447 SparseCPU, SparseCUDA: sgn_sparse_out
486 SparseCPU, SparseCUDA: conj_physical_out_sparse
552 SparseCPU, SparseCUDA, SparseMeta: add_sparse
564 SparseCPU, SparseCUDA, SparseMeta: add_sparse_
579 SparseCUDA: add_out_sparse_cuda
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | tensor_options_cuda.cpp | 41 options = getDeprecatedTypeProperties(Backend::SparseCUDA, kFloat).options(); in TEST()
44 options = getDeprecatedTypeProperties(Backend::SparseCUDA, kByte).options(); in TEST()
53 getDeprecatedTypeProperties(Backend::SparseCUDA, kFloat) in TEST()
62 options = empty(5, getDeprecatedTypeProperties(Backend::SparseCUDA, kByte)) in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_types.cpp | 40 case at::Backend::SparseCUDA: in backend_to_string()
158 Backend::CPU, Backend::CUDA, Backend::SparseCPU, Backend::SparseCUDA}; in all_declared_types()
175 (backend == Backend::SparseCUDA || backend == Backend::SparseCPU)) { in all_declared_types()
|
| H A D | tensor_new.cpp | 564 c10::DispatchKey::SparseCUDA, in check_base_legacy_new()
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 38 SparseCUDA: add_out_sparse_cuda
49 SparseCPU, SparseCUDA: add_sparse
64 SparseCUDA: mul_out_sparse_cuda
73 SparseCPU, SparseCUDA: mul_sparse
|
| /aosp_15_r20/external/pytorch/c10/test/core/ |
| H A D | DispatchKeySet_test.cpp | 174 // functionality bit + one backend bit. e.g. CPU, CUDA, SparseCPU, SparseCUDA,
327 {DispatchKey::Functionalize, DispatchKey::SparseCUDA}); in TEST()
329 DispatchKey::SparseCUDA, c10::highestPriorityBackendTypeId(sparse_cuda)); in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Ops.cpp | 524 // TODO: The SparseCPU/SparseCUDA dispatched methods are only used to support
530 TORCH_LIBRARY_IMPL(c10d, SparseCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 166 SparseCUDA = auto() variable in DispatchKey
282 DispatchKey.SparseCUDA,
319 DispatchKey.SparseCUDA,
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 249 (backend == at::Backend::CUDA || backend == at::Backend::SparseCUDA); in set_type()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | VariableTypeManual.cpp | 47 return allTypesForBackends({Backend::CUDA, Backend::SparseCUDA}); in allCUDATypes()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_opinfo.py | 253 }, # NYI: could not find kernel for aten.view.default at dispatch key DispatchKey.SparseCUDA
|