Searched defs:ExitGradientColocation (Results 1 – 2 of 2) sorted by relevance
470 def ExitGradientColocation(self, op: ops.Operation, gradient_uid: Text): member in TPUReplicateContext
702 def ExitGradientColocation(self, op, gradient_uid): member in ControlFlowContext