Searched refs:outer_grad_state (Results 1 – 2 of 2) sorted by relevance
122 def __init__(self, forward_ctxt, outer_grad_state): argument154 self._outer_grad_state = outer_grad_state155 if outer_grad_state:156 outer_forward_ctxt = outer_grad_state.forward_context167 cnt, forward_index = forward_ctxt.AddForwardLoopCounter(outer_grad_state)174 if outer_grad_state:178 history_cnt = outer_grad_state.AddForwardAccumulator(cnt)180 outer_grad_ctxt = outer_grad_state.grad_context189 real_cnt = outer_grad_state.AddBackpropAccumulatedValue(history_cnt, cnt)191 real_cnt, outer_grad_state)[all …]
1841 def AddForwardLoopCounter(self, outer_grad_state): argument1861 if outer_grad_state is not None:1864 outer_add_op = outer_grad_state.forward_index.op.inputs[0].op1890 def AddBackpropLoopCounter(self, count, outer_grad_state): argument1945 if outer_grad_state is not None:1949 outer_grad_state.grad_sync._add_control_input(final_zero.op)2000 outer_grad_state = self.grad_state.outer_grad_state2001 history_zeros_shape = outer_grad_state.AddForwardAccumulator(2004 real_shape = outer_grad_state.AddBackpropAccumulatedValue(