Searched refs:unused_exits (Results 1 – 2 of 2) sorted by relevance
690 grad_state.unused_exits.append(y)694 for y in grad_state.unused_exits:700 for y in grad_state.unused_exits:
946 def unused_exits(self): member in GradLoopState1233 grad_state.unused_exits.append(y)1235 loop_exits.extend(grad_state.unused_exits)