Searched refs:AddInnerOp (Results 1 – 2 of 2) sorted by relevance
170 def AddInnerOp(self, op): member in TPUReplicateContext173 self._outer_context.AddInnerOp(op)
932 self._grad_context.outer_context.AddInnerOp(self._grad_sync)1644 def AddInnerOp(self, op): member in ControlFlowContext1647 self._outer_context.AddInnerOp(op)1815 self._outer_context.AddInnerOp(result.op)1860 self._outer_context.AddInnerOp(op)2467 self._outer_context.AddInnerOp(enter.op)2545 self._outer_context.AddInnerOp(op)2899 self._outer_context.AddInnerOp(x.op)