Searched refs:unused_exits (Results 1 – 2 of 2) sorted by relevance
815 grad_state.unused_exits.append(y)819 for y in grad_state.unused_exits:825 for y in grad_state.unused_exits:
813 def unused_exits(self): member in GradLoopState1090 grad_state.unused_exits.append(y)1092 loop_exits.extend(grad_state.unused_exits)