Searched refs:setFuncTorchTLS (Results 1 – 3 of 3) sorted by relevance
18 void setFuncTorchTLS(const std::shared_ptr<const FuncTorchTLSBase>& state) { in setFuncTorchTLS() function
40 TORCH_API void setFuncTorchTLS(
59 functorch::setFuncTorchTLS(state.functorch_tls_); in setThreadLocalState()