Searched refs:FuncTorchDynamicLayerFrontMode (Results 1 – 13 of 13) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
H A D | Interpreter.cpp | 16 DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::FuncTorchDynamicLayerFrontMode) - in get_all_dynlayer_keyset() 18 result = result | DispatchKeySet({DispatchKey::FuncTorchDynamicLayerFrontMode}); in get_all_dynlayer_keyset()
|
H A D | DynamicLayer.cpp | 23 c10::impl::tls_set_dispatch_key_included(DispatchKey::FuncTorchDynamicLayerFrontMode, included); in setDynamicLayerFrontBackKeysIncluded() 493 TORCH_LIBRARY_IMPL(_, FuncTorchDynamicLayerFrontMode, m) { in TORCH_LIBRARY_IMPL() argument
|
H A D | TensorWrapper.cpp | 189 {DispatchKey::FuncTorchDynamicLayerFrontMode, in dead_tensor_wrapper_fallback()
|
H A D | ADInterpreters.cpp | 40 c10::impl::ExcludeDispatchKeyGuard guard(c10::DispatchKey::FuncTorchDynamicLayerFrontMode); in materializeGradWrappers()
|
H A D | PyTorchOperatorHacks.cpp | 258 TORCH_LIBRARY_IMPL(aten, FuncTorchDynamicLayerFrontMode, m) { in TORCH_LIBRARY_IMPL() argument
|
/aosp_15_r20/external/pytorch/c10/core/ |
H A D | DispatchKey.cpp | 181 case DispatchKey::FuncTorchDynamicLayerFrontMode: in toString() 311 c10::DispatchKey::FuncTorchDynamicLayerFrontMode}, in parseDispatchKey()
|
H A D | DispatchKey.h | 399 FuncTorchDynamicLayerFrontMode, // See Note [Out-of-tree vmap+grad prototype] enumerator
|
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
H A D | RegisterFunctionalization.cpp | 37 c10::DispatchKey::FuncTorchDynamicLayerFrontMode,
|
/aosp_15_r20/external/pytorch/torch/csrc/utils/ |
H A D | python_dispatch.cpp | 719 DEF_ONE(FuncTorchDynamicLayerFrontMode) in initDispatchBindings() 936 include_set.has(c10::DispatchKey::FuncTorchDynamicLayerFrontMode) || in initDispatchBindings()
|
H A D | tensor_new.cpp | 397 c10::DispatchKey::FuncTorchDynamicLayerFrontMode); in internal_new_from_data()
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _ops.py | 297 if dispatch_key == DispatchKey.FuncTorchDynamicLayerFrontMode:
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | init.cpp | 50 : front_guard_(c10::DispatchKey::FuncTorchDynamicLayerFrontMode), in DisableFuncTorch()
|
/aosp_15_r20/external/pytorch/torchgen/ |
H A D | model.py | 121 FuncTorchDynamicLayerFrontMode = auto() variable in DispatchKey
|