Searched refs:forward_loop_exits (Results 1 – 1 of 1) sorted by relevance
823 def forward_loop_exits(self): member in GradLoopState1086 for y in grad_state.forward_loop_exits:1133 for loop_exit in grad_state.forward_loop_exits: