Searched refs:tpu_compilation_device (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 525 string* tpu_compilation_device);
543 static Status BuildSequencingNodes(const string& tpu_compilation_device,
|
| D | distributed_tpu_rewrite_pass.cc | 1359 string* tpu_compilation_device, int* num_tpus_per_task, in GetTPUDeviceNames() argument
1371 *tpu_compilation_device = in GetTPUDeviceNames()
3922 string* tpu_compilation_device) { in GetDeviceTopology() argument
3935 device_set, tpu_compilation_device, in GetDeviceTopology()
4047 const string& tpu_compilation_device, const Node& replicate_node, in BuildSequencingNodes() argument
4074 predecessor->set_assigned_device_name(tpu_compilation_device); in BuildSequencingNodes()
4085 /*device=*/tpu_compilation_device, graph, control_after)); in BuildSequencingNodes()
4229 string tpu_compilation_device; in RewriteTPUReplicateNode() local
4233 &tpu_compilation_device)); in RewriteTPUReplicateNode()
4297 tpu_compilation_device, *replicate_node, graph, &host_transfer_sequencer, in RewriteTPUReplicateNode()
[all …]
|