Searched refs:exit_node (Results 1 – 4 of 4) sorted by relevance
| /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 | 390 Node* exit_node; in CreateExitNodes() local
400 .Finalize(graph_, &exit_node)); in CreateExitNodes()
401 exit_nodes_.emplace_back(exit_node); in CreateExitNodes()
402 outputs.emplace_back(NodeOut(exit_node, 0)); in CreateExitNodes()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 350 Node* exit_node; in AddNoOpAfterLastIteration() local
353 exit_node = out_node; in AddNoOpAfterLastIteration()
358 TF_RET_CHECK(exit_node != nullptr) in AddNoOpAfterLastIteration()
381 graph->AddControlEdge(after_last_iteration_node, exit_node); in AddNoOpAfterLastIteration()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 204 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
218 assert isinstance(exit_node, ops.Tensor)
219 self._while_context = exit_node.op._get_control_flow_context()
|