Searched refs:deferred_exits (Results 1 – 2 of 2) sorted by relevance
758 grad_state.deferred_exits.append(x)763 for y in grad_state.deferred_exits:
279 def deferred_exits(self): member in _GradLoopState