Searched refs:PythonCommHook (Results 1 – 3 of 3) sorted by relevance
10 PythonCommHook::~PythonCommHook() { in ~PythonCommHook()21 c10::intrusive_ptr<c10::ivalue::Future> PythonCommHook::runHook( in runHook()42 at::Tensor PythonCommHook::parseHookResult(const c10::IValue& result) { in parseHookResult()
12 class TORCH_PYTHON_API PythonCommHook : public CommHookInterface {19 PythonCommHook(py::object state, py::object hook) in PythonCommHook() function22 ~PythonCommHook() override;
362 reducer.register_comm_hook(std::make_unique<::c10d::PythonCommHook>( in _register_comm_hook()