Searched defs:set_tls_state (Results 1 – 2 of 2) sorted by relevance
19 void AutogradState::set_tls_state(AutogradState state) { in set_tls_state() function in c10::AutogradState
53 void SavedTensorDefaultHooks::set_tls_state(const impl::SavedTensorDefaultHooksTLS& state) { in set_tls_state() function in at::SavedTensorDefaultHooks