Lines Matching defs:exit
315 def exit(data, name=None): # pylint: disable=redefined-builtin
318 Exit makes its input `data` available to the parent frame.
336 values = exit(data.values, name=name)
841 outer_forward_ctxt.Exit()
864 outer_grad_ctxt.Exit()
878 outer_forward_ctxt.Exit()
1021 curr_ctxt.Exit()
1036 self.forward_context.Exit()
1051 value_ctxt.outer_context.Exit()
1059 value_ctxt.Exit()
1110 self.grad_context.Exit()
1159 self._grad_context.Exit()
1204 The "unused" exits of the loops are added to `unused_exits`. An exit is
1205 unused if its pending_count is 0. If there is an exit with real gradient,
1215 The exit node for x2 is not included by the betweenness analysis. But we
1250 """Exit the WhileContext for gradient computation."""
1253 Exit()
1258 Note that op is an Exit, and this method must be called in
1283 """Create zeros_like gradient for a loop exit.
1287 zero-valued tensor that is fed as gradient for the Exit node of that
1288 loop variable. Note that val.op is an Exit, and this method must be
1292 val: The output tensor of an Exit op.
1312 outer_grad_state.grad_context.Exit()
1317 forward_ctxt.outer_context.Exit()
1326 outer_grad_ctxt.Exit()
1379 op_ctxt.outer_context.Exit()
1385 op_ctxt.Exit()
1388 grad_state.grad_context.Exit()
1422 grad_state.grad_context.Exit()
1433 outer_grad_ctxt.Exit()
1437 grad_state.grad_context.Exit()
1486 The usage pattern is a sequence of (Enter, Exit) followed by a final
1492 construct new nodes. Changed by ctxt.Enter() and ctxt.Exit()
1498 Pushed and popped by ctxt.Enter() and ctxt.Exit()
1595 def Exit(self):
1596 """Exit this control flow context."""
1793 ret.Exit()
2055 context_t.Exit()
2064 context_f.Exit()
2256 # The list of exit tensors for loop variables.
2290 # The list of exit tensors for loop variables.
2346 """The list of exit tensors for loop variables."""
2422 ret.Exit()
2537 self.Exit()
2611 total_iterations = exit(switch_n[0], name="f_count")
2614 self.Exit()
2661 final_zero = exit(switch_count[0], name="b_count")
2671 self.Exit()
2696 self.Exit()
2708 self.outer_context.Exit()
2717 forward_ctxt.outer_context.Exit()
2725 self.outer_context.Exit()
2732 self.outer_context.Exit()
2751 result_acc = exit(switch_acc_false, name="b_acc")
2773 self.Exit()
2784 self.outer_context.Exit()
2798 self.outer_context.Exit()
2805 self.outer_context.Exit()
2849 exit_acc = [exit(x[0], name="b_acc") for x in switch_acc]
2986 # Add the exit ops.
2987 exit_vars = [exit(x[0]) for x in switch_vars]
2990 # Exit the loop.
3009 self.Exit()
