Searched refs:loop_exits (Results 1 – 3 of 3) sorted by relevance
827 self._forward_loop_exits = list(forward_ctxt.loop_exits)828 self._pending_exits_count = len(forward_ctxt.loop_exits)1226 loop_exits = []1235 loop_exits.extend(grad_state.unused_exits)1241 return loop_exits2345 def loop_exits(self): member in WhileContext2612 self.loop_exits.append(total_iterations)2662 self.loop_exits.append(final_zero)2752 self.loop_exits.append(result_acc)2850 self.loop_exits.extend(exit_acc)
226 grad_ctxt.loop_exits.append(result)
549 loop_exits = loop_state.ProcessUnusedLoopExits(pending_count, to_ops_set)550 for y in loop_exits: