Searched defs:tpu_compilation_device (Results 1 – 1 of 1) sorted by relevance
1359 string* tpu_compilation_device, int* num_tpus_per_task, in GetTPUDeviceNames()3922 string* tpu_compilation_device) { in GetDeviceTopology()4047 const string& tpu_compilation_device, const Node& replicate_node, in BuildSequencingNodes()4229 string tpu_compilation_device; in RewriteTPUReplicateNode() local