Searched refs:_hook_with_zero_step_setup (Results 1 – 1 of 1) sorted by relevance
132 def _hook_with_zero_step_setup( function265 _hook_with_zero_step_setup(ddp_ref, zero, bucket)424 _hook_with_zero_step_setup(ddp_ref, zero, bucket)