Home
last modified time | relevance | path

Searched refs:after_last_iteration_node (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc374 Node* after_last_iteration_node = in AddNoOpAfterLastIteration() local
380 graph->AddEdge(loop_switch_node, 0, after_last_iteration_node, 0); in AddNoOpAfterLastIteration()
381 graph->AddControlEdge(after_last_iteration_node, exit_node); in AddNoOpAfterLastIteration()
382 *node_out = after_last_iteration_node; in AddNoOpAfterLastIteration()
491 Node* after_last_iteration_node; in AddReshardOp() local
493 &after_last_iteration_node)); in AddReshardOp()
612 graph->AddControlEdge(after_last_iteration_node, unshard_op_node); in AddReshardOp()