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

  /external/tensorflow/tensorflow/python/distribute/
one_device_strategy.py 55 self._input_device = device_util.canonicalize("/device:CPU:0")
56 worker_device_pairs = [(self._input_device, [self._device])]
91 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
  /external/tensorflow/tensorflow/python/tpu/
tpu_context.py 53 self._input_device = input_device
101 return (self._input_device, self._invocation_index,

Completed in 167 milliseconds