Searched refs:PythonDispatcher (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dispatch.py | 10 from torch._python_dispatcher import PythonDispatcher
989 dispatcher = PythonDispatcher()
1010 dispatcher = PythonDispatcher()
1047 dispatcher = PythonDispatcher()
1085 dispatcher = PythonDispatcher()
1119 dispatcher = PythonDispatcher()
1125 dispatcher = PythonDispatcher()
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | PythonDispatcherTLS.cpp | 12 DispatchKey::PythonDispatcher, true); in set_state()
26 DispatchKey::PythonDispatcher, false); in reset_state()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | PythonFallbackKernel.cpp | 99 (*state)->python_dispatcher(op, dispatch_keys.remove(c10::DispatchKey::PythonDispatcher), stack); in pythonDispatcherFallback()
157 TORCH_LIBRARY_IMPL(_, PythonDispatcher, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 193 case DispatchKey::PythonDispatcher: in toString()
315 {"PythonDispatcher", c10::DispatchKey::PythonDispatcher}, in parseDispatchKey()
|
| H A D | DispatchKey.h | 425 PythonDispatcher, enumerator
|
| H A D | DispatchKeySet.h | 189 *this = add(DispatchKey::PythonDispatcher); in DispatchKeySet()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _python_dispatcher.py | 55 class PythonDispatcher: class
|
| H A D | _ops.py | 233 DispatchKey.PythonDispatcher, # type: ignore[attr-defined]
901 DispatchKey.PythonDispatcher,
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | delegate.py | 37 executorch_call_delegate.fallthrough(torch._C.DispatchKey.PythonDispatcher)
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | executorch_call_delegate.py | 37 executorch_call_delegate.fallthrough(torch._C.DispatchKey.PythonDispatcher)
|
| H A D | triton_kernel_wrap.py | 735 triton_kernel_wrapper_mutation.fallthrough(DispatchKey.PythonDispatcher) # type: ignore[attr-defin…
744 triton_kernel_wrapper_functional.fallthrough(DispatchKey.PythonDispatcher) # type: ignore[attr-def…
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 725 DEF_ONE(PythonDispatcher) in initDispatchBindings()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | README.md | 599 3. Validate the computed dispatch table matches what you want. You can use `PythonDispatcher` provi…
604 dispatcher = PythonDispatcher()
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 100 PythonDispatcher = auto() variable in DispatchKey
|