Searched refs:outer_grad_state (Results 1 – 1 of 1) sorted by relevance
661 def __init__(self, forward_ctxt, outer_grad_state): argument693 self._outer_grad_state = outer_grad_state694 if outer_grad_state:695 outer_forward_ctxt = outer_grad_state.forward_context706 cnt, forward_index = forward_ctxt.AddForwardLoopCounter(outer_grad_state)713 if outer_grad_state:717 history_cnt = outer_grad_state.AddForwardAccumulator(cnt)719 outer_grad_ctxt = outer_grad_state.grad_context728 real_cnt = outer_grad_state.AddBackpropAccumulatedValue(history_cnt, cnt)730 real_cnt, outer_grad_state)[all …]