Searched defs:tpu_device (Results 1 – 3 of 3) sorted by relevance
460 def generate_dequeue_op(self, tpu_device=0): argument767 def generate_dequeue_op(self, tpu_device=0): argument
165 def tpu_device(self, replica=0, logical_core=0, job=None): member in DeviceAssignment
92 DeviceProperties tpu_device; in TEST() local