Searched refs:AddInnerOp (Results 1 – 3 of 3) sorted by relevance
429 self._outer_context.AddInnerOp(op)448 def AddInnerOp(self, op): member in TPUReplicateContext451 self._outer_context.AddInnerOp(op)1470 self._outer_context.AddInnerOp(op)1478 def AddInnerOp(self, op): member in _TPUInferenceContext
242 self._outer_context.AddInnerOp(op)261 def AddInnerOp(self, op): member in XLACompileContext264 self._outer_context.AddInnerOp(op)
799 self._grad_context.outer_context.AddInnerOp(self._grad_sync)1529 def AddInnerOp(self, op): member in ControlFlowContext1532 self._outer_context.AddInnerOp(op)1695 self._outer_context.AddInnerOp(result.op)1772 self._outer_context.AddInnerOp(op)2437 self._outer_context.AddInnerOp(enter.op)2518 self._outer_context.AddInnerOp(op)2873 self._outer_context.AddInnerOp(x.op)