Searched defs:StashTorchDispatchModeGuard (Results 1 – 1 of 1) sorted by relevance
7 struct StashTorchDispatchModeGuard { struct9 StashTorchDispatchModeGuard() { in StashTorchDispatchModeGuard() function21 ~StashTorchDispatchModeGuard() { in ~StashTorchDispatchModeGuard() argument31 const std::shared_ptr<c10::impl::PyObject_TorchDispatchMode>& get_cur_mode() { in get_cur_mode()37 std::optional<c10::impl::TorchDispatchModeKey> saved_mode_key_;