Home
last modified time | relevance | path

Searched refs:PythonOnCompletionHook (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DPyProcessGroup.hpp208 class TORCH_PYTHON_API PythonOnCompletionHook { class
212 PythonOnCompletionHook(py::object hook) : hook_(std::move(hook)) {} in PythonOnCompletionHook() function in c10d::PythonOnCompletionHook
214 ~PythonOnCompletionHook() { in ~PythonOnCompletionHook()
H A Dinit.cpp2143 [hookWrapper = ::c10d::PythonOnCompletionHook(std::move( in c10d_init()