Searched refs:PythonTorchFunctionTLS (Results 1 – 10 of 10) sorted by relevance
6 static thread_local PythonTorchFunctionTLS pythonTorchFunctionState;8 void PythonTorchFunctionTLS::push_onto_stack(std::shared_ptr<SafePyObject> mode) { in push_onto_stack()12 const std::shared_ptr<SafePyObject> PythonTorchFunctionTLS::pop_stack() { in pop_stack()19 const std::shared_ptr<SafePyObject>& PythonTorchFunctionTLS::get_stack_at(int64_t idx) { in get_stack_at()24 int64_t PythonTorchFunctionTLS::stack_len() { in stack_len()28 void PythonTorchFunctionTLS::set_disabled_state(TorchFunctionDisabledState disabled_state) { in set_disabled_state()32 TorchFunctionDisabledState PythonTorchFunctionTLS::get_disabled_state() { in get_disabled_state()36 void PythonTorchFunctionTLS::set_state(const PythonTorchFunctionTLS& state) { in set_state()40 const PythonTorchFunctionTLS& PythonTorchFunctionTLS::get_state() { in get_state()45 return PythonTorchFunctionTLS::get_disabled_state() != TorchFunctionDisabledState::ALL_DISABLED && in torch_function_mode_enabled()[all …]
10 struct TORCH_API PythonTorchFunctionTLS { struct19 static const PythonTorchFunctionTLS& get_state();20 static void set_state(const PythonTorchFunctionTLS& state);
20 python_torch_function_state_(at::impl::PythonTorchFunctionTLS::get_state()), in ThreadLocalState()47 at::impl::PythonTorchFunctionTLS::set_state(state.python_torch_function_state_); in setThreadLocalState()
71 at::impl::PythonTorchFunctionTLS python_torch_function_state_;
14 return at::impl::PythonTorchFunctionTLS::get_disabled_state() == in torch_function_enabled()44 const auto old_state = at::impl::PythonTorchFunctionTLS::get_disabled_state(); in DisableTorchFunctionSubclass__enter()47 at::impl::PythonTorchFunctionTLS::set_disabled_state( in DisableTorchFunctionSubclass__enter()54 at::impl::PythonTorchFunctionTLS::set_disabled_state( in DisableTorchFunctionSubclass__exit()140 at::impl::PythonTorchFunctionTLS::get_disabled_state(); in DisableTorchFunction__enter()141 at::impl::PythonTorchFunctionTLS::set_disabled_state( in DisableTorchFunction__enter()147 at::impl::PythonTorchFunctionTLS::set_disabled_state( in DisableTorchFunction__exit()229 auto old_value = at::impl::PythonTorchFunctionTLS::get_disabled_state(); in THPModule_disable_torch_function()231 at::impl::PythonTorchFunctionTLS::set_disabled_state( in THPModule_disable_torch_function()236 at::impl::PythonTorchFunctionTLS::set_disabled_state(old_value); in THPModule_disable_torch_function()
9 cur_mode_ = at::impl::PythonTorchFunctionTLS::pop_stack(); in StashTorchFunctionModeGuard()12 at::impl::PythonTorchFunctionTLS::push_onto_stack(cur_mode_); in ~StashTorchFunctionModeGuard()
62 : old_(at::impl::PythonTorchFunctionTLS::get_disabled_state()) { in EnableTorchFunction()63 at::impl::PythonTorchFunctionTLS::set_disabled_state( in EnableTorchFunction()67 at::impl::PythonTorchFunctionTLS::set_disabled_state(old_); in ~EnableTorchFunction()1085 at::impl::PythonTorchFunctionTLS::push_onto_stack( in push_on_torch_function_stack()1096 const auto& mode = at::impl::PythonTorchFunctionTLS::pop_stack(); in pop_torch_function_stack()1114 const auto& mode = at::impl::PythonTorchFunctionTLS::get_stack_at(idx); in get_function_stack_at()1125 const auto len = at::impl::PythonTorchFunctionTLS::stack_len(); in len_torch_function_stack()
183 at::impl::PythonTorchFunctionTLS::push_onto_stack( in THPDevice_enter()194 at::impl::PythonTorchFunctionTLS::pop_stack(); in THPDevice_exit()
2546 int64_t len = at::impl::PythonTorchFunctionTLS::stack_len(); in check_nopybind()2551 at::impl::PythonTorchFunctionTLS::get_stack_at(idx); in check_nopybind()
1020 "aten/src/ATen/PythonTorchFunctionTLS.cpp",