Searched defs:tpu_compilation_device (Results 1 – 1 of 1) sorted by relevance
1568 string* tpu_compilation_device, int* num_tpus_per_task, in GetTPUDeviceNames()4377 string* tpu_compilation_device) { in GetDeviceTopology()4502 const string& tpu_compilation_device, const Node& replicate_node, in BuildSequencingNodes()4731 string tpu_compilation_device; in RewriteTPUReplicateNode() local