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