Searched refs:forward_loop_exits (Results 1 – 1 of 1) sorted by relevance
284 def forward_loop_exits(self): member in _GradLoopState540 for y in grad_state.forward_loop_exits:587 for loop_exit in grad_state.forward_loop_exits: