Searched refs:exit_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tfrt_graph_execution_state.cc | 506 const NodeDef& exit_node, in FindLoopCondFromExitNode() argument
509 for (const std::string& tensor_name : exit_node.input()) { in FindLoopCondFromExitNode()
513 " of exit node ", exit_node.name()); in FindLoopCondFromExitNode()
522 return errors::InvalidArgument("Exit node ", exit_node.name(), in FindLoopCondFromExitNode()
571 for (const NodeDef* exit_node : exit_nodes) { in PruneGraphDef() local
573 FindLoopCondFromExitNode(*exit_node, name_to_node)); in PruneGraphDef()
574 loop_cond_to_exit_nodes[loop_cond_node].insert(exit_node); in PruneGraphDef()
640 for (const NodeDef* exit_node : loop_cond_to_exit_nodes[node]) { in PruneGraphDef() local
641 queue.push_back(exit_node); in PruneGraphDef()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 93 Status ProcessWhileLoop(Node* exit_node, const Output& summed_grads);
391 Status SymbolicGradientBuilder::ProcessWhileLoop(Node* exit_node, in ProcessWhileLoop() argument
402 DCHECK(exit_node->IsExit()); in ProcessWhileLoop()
403 WhileContext* while_ctx = exit_node->while_ctx(); in ProcessWhileLoop()
408 DCHECK(backprops.find(exit_node) == backprops.end()); in ProcessWhileLoop()
409 backprops[exit_node] = summed_grads; in ProcessWhileLoop()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 393 Node* exit_node; in CreateExitNodes() local
402 .Finalize(graph_, &exit_node)); in CreateExitNodes()
403 exit_nodes_.emplace_back(exit_node); in CreateExitNodes()
404 outputs.emplace_back(NodeOut(exit_node, 0)); in CreateExitNodes()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
225 assert isinstance(exit_node, ops.Tensor)
226 self._while_context = exit_node.op._get_control_flow_context()
|