Searched refs:EnterGradWhileContext (Results 1 – 2 of 2) sorted by relevance
592 loop_state.EnterGradWhileContext(op, before=True)640 loop_state.EnterGradWhileContext(op, before=False)
553 def EnterGradWhileContext(self, op, before): member in _ControlFlowState