Searched refs:apply_optim_in_backward_hook (Results 1 – 1 of 1) sorted by relevance
64 def apply_optim_in_backward_hook( function119 apply_optim_in_backward_hook, optim_stream_state=optim_in_bwd_state122 comm_hook.__name__ = apply_optim_in_backward_hook.__name__123 comm_hook.__qualname__ = apply_optim_in_backward_hook.__qualname__