Searched refs:AddInnerOp (Results 1 – 4 of 4) sorted by relevance
271 self._outer_context.AddInnerOp(op)290 def AddInnerOp(self, op): member in XLACompileContext293 self._outer_context.AddInnerOp(op)
646 self._outer_context.AddInnerOp(op)668 def AddInnerOp(self, op: ops.Operation): member in TPUReplicateContext671 self._outer_context.AddInnerOp(op)717 def AddInnerOp(self, op: ops.Operation) -> None: member in OutsideCompilationV2Context719 self._outer_context.AddInnerOp(op)2126 self._outer_context.AddInnerOp(op)2134 def AddInnerOp(self, op): member in _TPUInferenceContext
797 def AddInnerOp(self, op): member in ControlFlowContext800 self._outer_context.AddInnerOp(op)963 self._outer_context.AddInnerOp(result.op)1040 self._outer_context.AddInnerOp(op)1728 self._outer_context.AddInnerOp(enter.op)1818 self._outer_context.AddInnerOp(op)2173 self._outer_context.AddInnerOp(x.op)
260 self._grad_context.outer_context.AddInnerOp(self._grad_sync)