Searched defs:set_tls_state (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/c10/core/ | ||
| D | AutogradState.cpp | 19 void AutogradState::set_tls_state(AutogradState state) { in set_tls_state() function in c10::AutogradState |
| /external/pytorch/aten/src/ATen/ | ||
| D | SavedTensorHooks.cpp | 53 void SavedTensorDefaultHooks::set_tls_state(const impl::SavedTensorDefaultHooksTLS& state) { in set_tls_state() function in at::SavedTensorDefaultHooks |