Home
last modified time | relevance | path

Searched refs:saved_tensors_default_hooks_state_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.cpp21saved_tensors_default_hooks_state_(at::SavedTensorDefaultHooks::get_tls_state()), functionalizatio… in ThreadLocalState()
51 at::SavedTensorDefaultHooks::set_tls_state(state.saved_tensors_default_hooks_state_); in setThreadLocalState()
H A DThreadLocalState.h74 at::impl::SavedTensorDefaultHooksTLS saved_tensors_default_hooks_state_; variable