Home
last modified time | relevance | path

Searched refs:FuncTorchDynamicLayerFrontMode (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DInterpreter.cpp16 DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::FuncTorchDynamicLayerFrontMode) - in get_all_dynlayer_keyset()
18 result = result | DispatchKeySet({DispatchKey::FuncTorchDynamicLayerFrontMode}); in get_all_dynlayer_keyset()
H A DDynamicLayer.cpp23 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 DTensorWrapper.cpp189 {DispatchKey::FuncTorchDynamicLayerFrontMode, in dead_tensor_wrapper_fallback()
H A DADInterpreters.cpp40 c10::impl::ExcludeDispatchKeyGuard guard(c10::DispatchKey::FuncTorchDynamicLayerFrontMode); in materializeGradWrappers()
H A DPyTorchOperatorHacks.cpp258 TORCH_LIBRARY_IMPL(aten, FuncTorchDynamicLayerFrontMode, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.cpp181 case DispatchKey::FuncTorchDynamicLayerFrontMode: in toString()
311 c10::DispatchKey::FuncTorchDynamicLayerFrontMode}, in parseDispatchKey()
H A DDispatchKey.h399 FuncTorchDynamicLayerFrontMode, // See Note [Out-of-tree vmap+grad prototype] enumerator
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DRegisterFunctionalization.cpp37 c10::DispatchKey::FuncTorchDynamicLayerFrontMode,
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp719 DEF_ONE(FuncTorchDynamicLayerFrontMode) in initDispatchBindings()
936 include_set.has(c10::DispatchKey::FuncTorchDynamicLayerFrontMode) || in initDispatchBindings()
H A Dtensor_new.cpp397 c10::DispatchKey::FuncTorchDynamicLayerFrontMode); in internal_new_from_data()
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py297 if dispatch_key == DispatchKey.FuncTorchDynamicLayerFrontMode:
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinit.cpp50 : front_guard_(c10::DispatchKey::FuncTorchDynamicLayerFrontMode), in DisableFuncTorch()
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py121 FuncTorchDynamicLayerFrontMode = auto() variable in DispatchKey