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