Searched defs:tpu_compilation_device (Results 1 – 1 of 1) sorted by relevance
1622 string* tpu_compilation_device, int* num_tpus_per_task, in GetTPUDeviceNames()4430 string* tpu_compilation_device) { in GetDeviceTopology()4555 const string& tpu_compilation_device, const Node& replicate_node, in BuildSequencingNodes()4788 string tpu_compilation_device; in RewriteTPUReplicateNode() local