Searched defs:dispatch_keys (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | PythonFallbackKernel.cpp | 96 void pythonDispatcherFallback(const c10::OperatorHandle& op, c10::DispatchKeySet dispatch_keys, tor… in pythonDispatcherFallback()
102 void pythonTLSSnapshotFallback(const c10::OperatorHandle &op, c10::DispatchKeySet dispatch_keys, to… in pythonTLSSnapshotFallback()
117 void preDispatchFallback(const c10::OperatorHandle& op, c10::DispatchKeySet dispatch_keys, torch::j… in preDispatchFallback()
|
| H A D | MetaFallbackKernel.cpp | 9 c10::DispatchKeySet dispatch_keys, in metaFallback()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd_not_implemented_fallback.cpp | 100 c10::DispatchKeySet dispatch_keys, in basicAutogradNotImplementedFallbackImpl()
227 c10::DispatchKeySet dispatch_keys, in basic_autograd_not_implemented_fallback()
234 c10::DispatchKeySet dispatch_keys, in autogradNotImplementedFallbackImpl()
480 c10::DispatchKeySet dispatch_keys, in autogradNotImplementedInplaceOrViewFallbackImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ConjugateFallback.cpp | 12 static void conjugateFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch::j… in conjugateFallback()
|
| H A D | ZeroTensorFallback.cpp | 14 …static void zeroTensorFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch:… in zeroTensorFallback()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NegateFallback.cpp | 13 static void negationFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch::ji… in negationFallback()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration.cpp | 93 std::unordered_set<DispatchKey> dispatch_keys; in checkNoDuplicateKernels_() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dispatch.py | 44 def extract_dispatch_table_with_keys(table, dispatch_keys): argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 274 dispatch_keys = [ variable
|