| /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_
578 SparseCPU, SparseMeta: add_out_sparse_cpu
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 37 SparseCPU, enumerator
93 } else if (t == DispatchKey::SparseCPU) { in dispatchKeyToBackend()
94 return Backend::SparseCPU; in dispatchKeyToBackend()
171 case Backend::SparseCPU: in backendToDispatchKey()
172 return DispatchKey::SparseCPU; in backendToDispatchKey()
224 case Backend::SparseCPU: in backendToDeviceType()
308 case Backend::SparseCPU: in toString()
309 return "SparseCPU"; in toString()
362 case Backend::SparseCPU: in isSparse()
|
| H A D | DispatchKey.cpp | 342 {"SparseCPU", c10::DispatchKey::SparseCPU}, in parseDispatchKey()
|
| H A D | DispatchKey.h | 80 // - sparse kernels (e.g. DispatchKey::SparseCPU)
120 // functionalities (e.g. CPU, SparseCPU, AutogradCPU) (5) alias keys (e.g.
544 // DispatchKeySet, while `SparseCPU`, `SparseCUDA`, etc all map to individual
|
| H A D | DispatchKeySet.h | 81 // - "Sparse": SparseCPU, SparseCUDA, ...
101 // SparseCPU, AutogradCPU)
463 case DispatchKey::SparseCPU: in getDispatchTableIndexForDispatchKeySet()
|
| H A D | Layout.h | 33 case Backend::SparseCPU: in layout_from_backend()
|
| H A D | TensorImpl.h | 2026 * CPU is compatible with CUDA and SparseCPU is
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | tensor_options.cpp | 62 TensorOptions(getDeprecatedTypeProperties(Backend::SparseCPU, kFloat)); in TEST()
66 TensorOptions(getDeprecatedTypeProperties(Backend::SparseCPU, kByte)); in TEST()
74 options = empty(5, getDeprecatedTypeProperties(Backend::SparseCPU, kByte)) in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_types.cpp | 38 case at::Backend::SparseCPU: 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 | 563 c10::DispatchKey::SparseCPU, in check_base_legacy_new()
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 37 SparseCPU: add_out_sparse_cpu
49 SparseCPU, SparseCUDA: add_sparse
63 SparseCPU: mul_out_sparse_cpu
73 SparseCPU, SparseCUDA: mul_sparse
|
| /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
526 TORCH_LIBRARY_IMPL(c10d, SparseCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 1417 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU)); in TEST()
1423 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU, /*requires_grad=*/true)); in TEST()
1596 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU)); in TEST()
1603 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU, /*requires_grad=*/true)); in TEST()
1650 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU)); in TEST()
1657 // AutogradOther is fallthrough, calls SparseCPU kernel in TEST()
1658 callOp(*op, dummyTensor(c10::DispatchKey::SparseCPU, /*requires_grad=*/true)); in TEST()
|
| /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,
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | VariableTypeManual.cpp | 42 return allTypesForBackends({Backend::CPU, Backend::SparseCPU}); in allCPUTypes()
|
| /aosp_15_r20/external/pytorch/ |
| H A D | pt_ops.bzl | 11 "SparseCPU", # brings ~20 kb size regression
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_opinfo.py | 234 }, # NYI: could not find kernel for aten.view.default at dispatch key DispatchKey.SparseCPU
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 165 SparseCPU = auto() variable in DispatchKey
276 DispatchKey.SparseCPU,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 19779 … # NotImplementedError: Could not run 'aten::normal_' with arguments from the 'SparseCPU' backend
|