Searched refs:tpu_device_assignment (Results 1 – 1 of 1) sorted by relevance
27 from tensorflow.python.tpu import device_assignment as tpu_device_assignment unknown288 device_assignment = tpu_device_assignment.device_assignment(