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