Searched refs:exit_node (Results 1 – 3 of 3) sorted by relevance
92 Status ProcessWhileLoop(Node* exit_node, const Output& summed_grads);390 Status SymbolicGradientBuilder::ProcessWhileLoop(Node* exit_node, in ProcessWhileLoop() argument401 DCHECK(exit_node->IsExit()); in ProcessWhileLoop()402 WhileContext* while_ctx = exit_node->while_ctx(); in ProcessWhileLoop()407 DCHECK(backprops.find(exit_node) == backprops.end()); in ProcessWhileLoop()408 backprops[exit_node] = summed_grads; in ProcessWhileLoop()
316 Node* exit_node; in CreateExitNodes() local321 .Finalize(graph_, &exit_node)); in CreateExitNodes()322 exit_nodes_.emplace_back(exit_node); in CreateExitNodes()323 outputs.emplace_back(NodeOut(exit_node, 0)); in CreateExitNodes()
98 def __init__(self, exit_node, pfor_ops, pfor_config): argument109 assert isinstance(exit_node, ops.Tensor)110 self._while_context = exit_node.op._get_control_flow_context()