Searched refs:unused_exits (Results 1 – 2 of 2) sorted by relevance
270 def unused_exits(self): member in _GradLoopState540 grad_state.unused_exits.append(y)542 loop_exits.extend(grad_state.unused_exits)
779 grad_state.unused_exits.append(y)783 for y in grad_state.unused_exits:789 for y in grad_state.unused_exits: