Home
last modified time | relevance | path

Searched defs:functorch_tls (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DDynamicLayer.cpp143 auto* functorch_tls = getRawFunctorchTLS(); in setInplaceRequiresGradAllowed() local
148 auto* functorch_tls = getRawFunctorchTLS(); in getInplaceRequiresGradAllowed() local
153 auto* functorch_tls = getRawFunctorchTLS(); in setSingleLevelAutogradFunctionAllowed() local
158 auto* functorch_tls = getRawFunctorchTLS(); in getSingleLevelAutogradFunctionAllowed() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dvariable.cpp535 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in retain_grad() local
H A Dpython_function.cpp1293 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in THPFunction_apply() local
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp830 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in THPVariable_requires_grad_() local