Searched refs:forward_sync (Results 1 – 1 of 1) sorted by relevance
224 def forward_sync(self): member in _GradLoopState382 self.forward_sync._add_control_input(push.op)