Searched refs:ThreadLocalPythonObjects (Results 1 – 6 of 6) sorted by relevance
9 static thread_local ThreadLocalPythonObjects py_objects;12 void ThreadLocalPythonObjects::set(const std::string& key, std::shared_ptr<SafePyObject> value) { in set()16 const std::shared_ptr<SafePyObject>& ThreadLocalPythonObjects::get(const std::string& key) { in get()21 bool ThreadLocalPythonObjects::contains(const std::string& key) { in contains()25 void ThreadLocalPythonObjects::set_state(ThreadLocalPythonObjects state) { in set_state()29 const ThreadLocalPythonObjects& ThreadLocalPythonObjects::get_state() { in get_state()
9 struct TORCH_API ThreadLocalPythonObjects { struct14 static const ThreadLocalPythonObjects& get_state();15 static void set_state(ThreadLocalPythonObjects state);
22 saved_objects_(at::impl::ThreadLocalPythonObjects::get_state()) { in ThreadLocalState()63 at::impl::ThreadLocalPythonObjects::set_state(state.saved_objects_); in setThreadLocalState()
79 at::impl::ThreadLocalPythonObjects saved_objects_;
2088 at::impl::ThreadLocalPythonObjects::get_state().set( in initModule()2095 at::impl::ThreadLocalPythonObjects::get_state().get(key); in initModule()2101 return at::impl::ThreadLocalPythonObjects::get_state().contains(key); in initModule()
1021 "aten/src/ATen/ThreadLocalPythonObjects.cpp",