HomeSort by relevance Sort by last modified time
    Searched refs:tpu_device (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_placer_test.cc 92 DeviceProperties tpu_device; local
93 tpu_device.set_type("TPU");
94 devices["/job:localhost/replica:0/task:0/device:TPU:0"] = tpu_device;
  /external/tensorflow/tensorflow/python/tpu/
tpu_feed.py 455 def generate_dequeue_op(self, tpu_device=0):
463 tpu_device: The TPU device ordinal where the infeed instruction should be
486 if tpu_device is not None:
487 with ops.device(tpu.core(tpu_device)):
762 def generate_dequeue_op(self, tpu_device=0):
766 tpu_device: The TPU device ordinal where the infeed instruction should be
786 with ops.device(tpu.core(tpu_device)):
    [all...]
device_assignment.py 163 def tpu_device(self, replica=0, logical_core=0, job=None): member in class:DeviceAssignment
tpu_context.py 556 return self.device_assignment.tpu_device(replica=i, job=master)
tpu_estimator.py     [all...]
  /external/tensorflow/tensorflow/python/distribute/
tpu_strategy.py 242 for tpu_device in self._tpu_devices:
243 host_device = _get_host_for_device(tpu_device)
245 input_worker_devices[host_device].append(tpu_device)

Completed in 611 milliseconds