Searched refs:run_allreduce_hook (Results 1 – 3 of 3) sorted by relevance
113 c10::intrusive_ptr<c10::ivalue::Future> run_allreduce_hook(
936 return run_allreduce_hook(grad_bucket); in run_comm_hook()942 c10::intrusive_ptr<c10::ivalue::Future> Reducer::run_allreduce_hook( in run_allreduce_hook() function in c10d::Reducer
658 reducer.run_allreduce_hook(bucket); in c10d_init()