Searched defs:comm_hook (Results 1 – 4 of 4) sorted by relevance
19 def _ddp_comm_hook_wrapper(comm_hook, model, state): argument23 def _powerSGD_comm_hook_wrapper( argument
867 def comm_hook( function895 def comm_hook(state: object, bucket: dist.GradBucket) -> int: function907 def comm_hook(state: object, bucket: dist.GradBucket): function
2039 def comm_hook( function2067 def comm_hook(state: object, bucket: dist.GradBucket) -> int: function2079 def comm_hook(state: object, bucket: dist.GradBucket): function
361 py::object comm_hook) { in _register_comm_hook()