Searched refs:GetControlPivot (Results 1 – 2 of 2) sorted by relevance
1534 def GetControlPivot(self): member in ControlFlowContext1634 def GetControlPivot(self): member in CondContext2390 def GetControlPivot(self): member in WhileContext2483 op._add_control_input(self.GetControlPivot().op)2538 op._add_control_input(self.GetControlPivot().op)2879 control_pivot = outer_context.GetControlPivot()
395 op._add_control_input(self.GetControlPivot())468 def GetControlPivot(self): member in TPUReplicateContext