Searched refs:PythonOnCompletionHook (Results 1 – 2 of 2) sorted by relevance
208 class TORCH_PYTHON_API PythonOnCompletionHook { class212 PythonOnCompletionHook(py::object hook) : hook_(std::move(hook)) {} in PythonOnCompletionHook() function in c10d::PythonOnCompletionHook214 ~PythonOnCompletionHook() { in ~PythonOnCompletionHook()
2143 [hookWrapper = ::c10d::PythonOnCompletionHook(std::move( in c10d_init()