Searched refs:grad_sync (Results 1 – 2 of 2) sorted by relevance
248 def grad_sync(self): member in _GradLoopState434 self.grad_sync._add_control_input(pop.op)
1949 outer_grad_state.grad_sync._add_control_input(final_zero.op)