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