Searched refs:control_after (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 538 Node* control_after, absl::Span<const VariableInput> variable_nodes,
547 Node** control_after);
|
| D | distributed_tpu_rewrite_pass.cc | 4049 Node** control_after) { in BuildSequencingNodes() argument
4085 /*device=*/tpu_compilation_device, graph, control_after)); in BuildSequencingNodes()
4088 graph->AddControlEdge(successor, *control_after); in BuildSequencingNodes()
4090 graph->AddControlEdge(*control_after, successor); in BuildSequencingNodes()
4099 Node* control_after, absl::Span<const VariableInput> variable_nodes, in DealWithConstantsAndVariables() argument
4109 graph->AddControlEdge(host_transfer_sequencer, control_after); in DealWithConstantsAndVariables()
4295 Node* control_after; in RewriteTPUReplicateNode() local
4298 &control_before, &control_after)); in RewriteTPUReplicateNode()
4309 control_before, control_after, variable_inputs, in RewriteTPUReplicateNode()
4341 control_after_compilation, control_after, &variable_writes, graph)); in RewriteTPUReplicateNode()
[all …]
|