Searched refs:EnterGradientColocation (Results 1 – 3 of 3) sorted by relevance
212 def EnterGradientColocation(self, op, gradient_uid): member in TPUReplicateContext
1475 def EnterGradientColocation(self, op, gradient_uid): member in ControlFlowContext1478 self._outer_context.EnterGradientColocation(op, gradient_uid)
4263 self._control_flow_context.EnterGradientColocation(op, gradient_uid)