Searched refs:_RemoveExternalControlEdges (Results 1 – 3 of 3) sorted by relevance
147 def _RemoveExternalControlEdges(self, op): member in XLACompileContext202 external_control_inputs) = self._RemoveExternalControlEdges(op)
333 def _RemoveExternalControlEdges(self, op): member in TPUReplicateContext389 external_control_inputs) = self._RemoveExternalControlEdges(op)
1506 def _RemoveExternalControlEdges(self, op): member in ControlFlowContext1722 self._RemoveExternalControlEdges(op)1753 self._RemoveExternalControlEdges(op)2479 control_inputs, external_inputs = self._RemoveExternalControlEdges(op)2494 _, external_inputs = self._RemoveExternalControlEdges(op)