Searched refs:PythonTLSSnapshot (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | TorchDispatchModeTLS.cpp | 32 DispatchKey::PythonTLSSnapshot, true); in push_non_infra_mode_onto_stack()
60 DispatchKey::PythonTLSSnapshot, false); in pop_stack()
77 DispatchKey::PythonTLSSnapshot, false); in pop_highest_infra_mode()
142 DispatchKey::PythonTLSSnapshot, true); in set_mode()
156 DispatchKey::PythonTLSSnapshot, false); in unset_mode()
170 DispatchKey::PythonTLSSnapshot, false); in set_state()
174 DispatchKey::PythonTLSSnapshot, true); in set_state()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | PythonFallbackKernel.cpp | 108 … c10::DispatchKeySet(c10::DispatchKeySet::FULL_AFTER, c10::DispatchKey::PythonTLSSnapshot), stack); in pythonTLSSnapshotFallback()
161 TORCH_LIBRARY_IMPL(_, PythonTLSSnapshot, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | TorchDispatchUtils.cpp | 7 DispatchKeySet key_set({DispatchKey::Python, DispatchKey::PythonTLSSnapshot}); in tensor_has_dispatch()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 172 case DispatchKey::PythonTLSSnapshot: in toString()
281 {"PythonTLSSnapshot", c10::DispatchKey::PythonTLSSnapshot}, in parseDispatchKey()
|
| H A D | DispatchKeySet.h | 689 DispatchKey::PythonTLSSnapshot,
918 DispatchKey::PythonTLSSnapshot, in legacyExtractDispatchKey()
|
| H A D | DispatchKey.h | 396 PythonTLSSnapshot, enumerator
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 84 at::DispatchKey::PythonTLSSnapshot)) { in EnableHermeticPyObject()
89 at::DispatchKey::PythonTLSSnapshot, false); in EnableHermeticPyObject()
98 at::DispatchKey::PythonTLSSnapshot, old_python_snapshot_); in ~EnableHermeticPyObject()
717 DEF_ONE(PythonTLSSnapshot) in initDispatchBindings()
|
| H A D | disable_torch_function.h | 25 guard_tls_snapshot_(c10::DispatchKey::PythonTLSSnapshot) {}
|
| H A D | tensor_new.cpp | 393 c10::DispatchKey::PythonTLSSnapshot); in internal_new_from_data()
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | delegate.py | 38 executorch_call_delegate.fallthrough(torch._C.DispatchKey.PythonTLSSnapshot)
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | executorch_call_delegate.py | 38 executorch_call_delegate.fallthrough(torch._C.DispatchKey.PythonTLSSnapshot)
|
| H A D | triton_kernel_wrap.py | 736 triton_kernel_wrapper_mutation.fallthrough(DispatchKey.PythonTLSSnapshot) # type: ignore[attr-defi…
745 triton_kernel_wrapper_functional.fallthrough(DispatchKey.PythonTLSSnapshot) # type: ignore[attr-de…
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _ops.py | 234 DispatchKey.PythonTLSSnapshot, # type: ignore[attr-defined]
900 DispatchKey.PythonTLSSnapshot,
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 98 PythonTLSSnapshot = auto() variable in DispatchKey
|