Searched refs:functorch_tls (Results 1 – 5 of 5) sorted by relevance
143 auto* functorch_tls = getRawFunctorchTLS(); in setInplaceRequiresGradAllowed() local144 functorch_tls->allow_inplace_requires_grad_ = allowed; in setInplaceRequiresGradAllowed()148 auto* functorch_tls = getRawFunctorchTLS(); in getInplaceRequiresGradAllowed() local149 return functorch_tls->allow_inplace_requires_grad_; in getInplaceRequiresGradAllowed()153 auto* functorch_tls = getRawFunctorchTLS(); in setSingleLevelAutogradFunctionAllowed() local154 functorch_tls->allow_single_level_autograd_function_ = allowed; in setSingleLevelAutogradFunctionAllowed()158 auto* functorch_tls = getRawFunctorchTLS(); in getSingleLevelAutogradFunctionAllowed() local159 return functorch_tls->allow_single_level_autograd_function_; in getSingleLevelAutogradFunctionAllowed()
321 const auto& functorch_tls = at::functorch::functorchTLSAccessor();322 if (functorch_tls) {326 functorch_tls->checkSupportsCppAutogradFunction();
535 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in retain_grad() local536 if (functorch_tls) { in retain_grad()537 functorch_tls->checkSupportsRetainGrad(); in retain_grad()
1293 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in THPFunction_apply() local1294 if (functorch_tls) { in THPFunction_apply()1301 functorch_tls->checkSupportsSingleLevelAutogradFunction(); in THPFunction_apply()
830 const auto& functorch_tls = at::functorch::functorchTLSAccessor(); in THPVariable_requires_grad_() local831 if (functorch_tls) { in THPVariable_requires_grad_()832 functorch_tls->checkSupportsInplaceRequiresGrad(); in THPVariable_requires_grad_()