Searched refs:forward_ctxt (Results 1 – 1 of 1) sorted by relevance
661 def __init__(self, forward_ctxt, outer_grad_state): argument690 self._forward_loop_exits = list(forward_ctxt.loop_exits)691 self._pending_exits_count = len(forward_ctxt.loop_exits)697 if not hasattr(forward_ctxt, "outer_context"):700 outer_forward_ctxt = forward_ctxt.outer_context703 with forward_ctxt._graph.as_default(): # pylint: disable=protected-access706 cnt, forward_index = forward_ctxt.AddForwardLoopCounter(outer_grad_state)709 self._forward_context = forward_ctxt722 maximum_iterations=forward_ctxt.maximum_iterations,723 parallel_iterations=forward_ctxt.parallel_iterations,[all …]