Searched refs:forward_ctxt (Results 1 – 2 of 2) sorted by relevance
122 def __init__(self, forward_ctxt, outer_grad_state): argument151 self._forward_loop_exits = list(forward_ctxt.loop_exits)152 self._pending_exits_count = len(forward_ctxt.loop_exits)158 if not hasattr(forward_ctxt, "outer_context"):161 outer_forward_ctxt = forward_ctxt.outer_context164 with forward_ctxt._graph.as_default(): # pylint: disable=protected-access167 cnt, forward_index = forward_ctxt.AddForwardLoopCounter(outer_grad_state)170 self._forward_context = forward_ctxt183 maximum_iterations=forward_ctxt.maximum_iterations,184 parallel_iterations=forward_ctxt.parallel_iterations,[all …]
1707 forward_ctxt = util.GetWhileContext(val.op)1709 forward_ctxt = forward_ctxt.outer_context1710 if forward_ctxt:1711 forward_ctxt = forward_ctxt.GetWhileContext()1712 if forward_ctxt == grad_ctxt.grad_state.forward_context:1996 forward_ctxt = self.grad_state.forward_context1997 forward_ctxt.outer_context.Enter()1999 forward_ctxt.outer_context.Exit()