Searched refs:pending_exits_count (Results 1 – 2 of 2) sorted by relevance
806 grad_state.pending_exits_count -= 1807 if grad_state.pending_exits_count == 0:
828 def pending_exits_count(self): member in GradLoopState832 @pending_exits_count.setter833 def pending_exits_count(self, cnt): member in GradLoopState1088 grad_state.pending_exits_count -= 11091 if grad_state.pending_exits_count == 0: