Searched defs:PythonOnCompletionHook (Results 1 – 1 of 1) sorted by relevance
208 class TORCH_PYTHON_API PythonOnCompletionHook { class212 PythonOnCompletionHook(py::object hook) : hook_(std::move(hook)) {} in PythonOnCompletionHook() function in c10d::PythonOnCompletionHook