Searched refs:deferred_exits (Results 1 – 2 of 2) sorted by relevance
805 grad_state.deferred_exits.append(x)810 for y in grad_state.deferred_exits:
818 def deferred_exits(self): member in GradLoopState