Searched refs:forward_loop_exits (Results 1 – 1 of 1) sorted by relevance
280 def forward_loop_exits(self): member in _GradLoopState536 for y in grad_state.forward_loop_exits:583 for loop_exit in grad_state.forward_loop_exits: