Searched refs:EnterGradWhileContext (Results 1 – 2 of 2) sorted by relevance
657 loop_state.EnterGradWhileContext(op, before=True)691 loop_state.EnterGradWhileContext(op, before=False)
1099 def EnterGradWhileContext(self, op, before): member in ControlFlowState