Searched refs:GetControlPivot (Results 1 – 2 of 2) sorted by relevance
749 def GetControlPivot(self): member in ControlFlowContext849 def GetControlPivot(self): member in CondContext1639 def GetControlPivot(self): member in WhileContext1734 op._add_control_input(self.GetControlPivot().op) # pylint: disable=protected-access1741 op._add_control_input(self.GetControlPivot().op)1796 op._add_control_input(self.GetControlPivot().op)2152 control_pivot = outer_context.GetControlPivot()
635 op._add_control_input(self.GetControlPivot())711 def GetControlPivot(self) -> ops.Operation: member in TPUReplicateContext