Searched refs:host_transfer_sequencer (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 537 Graph* graph, Node* host_transfer_sequencer, Node* control_before,
545 Node** host_transfer_sequencer,
|
| D | distributed_tpu_rewrite_pass.cc | 4048 Graph* graph, Node** host_transfer_sequencer, Node** control_before, in BuildSequencingNodes() argument
4050 *host_transfer_sequencer = nullptr; in BuildSequencingNodes()
4067 if (*host_transfer_sequencer != nullptr) { in BuildSequencingNodes()
4070 (*host_transfer_sequencer)->name(), " and ", in BuildSequencingNodes()
4075 *host_transfer_sequencer = predecessor; in BuildSequencingNodes()
4098 Graph* graph, Node* host_transfer_sequencer, Node* control_before, in DealWithConstantsAndVariables() argument
4108 if (host_transfer_sequencer != nullptr) { in DealWithConstantsAndVariables()
4109 graph->AddControlEdge(host_transfer_sequencer, control_after); in DealWithConstantsAndVariables()
4293 Node* host_transfer_sequencer; in RewriteTPUReplicateNode() local
4297 tpu_compilation_device, *replicate_node, graph, &host_transfer_sequencer, in RewriteTPUReplicateNode()
[all …]
|