Searched refs:control_after_compilation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 371 Node** control_after_compilation, Graph* graph);
|
| D | distributed_tpu_rewrite_pass.cc | 4116 Node* replicate_node, Node* compile_node, Node** control_after_compilation, in BuildCompilationStatusReturnNodes() argument
4169 /*device=*/"", graph, control_after_compilation)); in BuildCompilationStatusReturnNodes()
4170 graph->AddControlEdge(compile_succeeded, *control_after_compilation); in BuildCompilationStatusReturnNodes()
4332 Node* control_after_compilation; in RewriteTPUReplicateNode() local
4334 replicate_node, compile_node, &control_after_compilation, graph)); in RewriteTPUReplicateNode()
4341 control_after_compilation, control_after, &variable_writes, graph)); in RewriteTPUReplicateNode()
|