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

  /external/tensorflow/tensorflow/python/framework/
device_test.py 29 d = device.DeviceSpec()
35 d = device.DeviceSpec(job="j", replica=0, task=1,
44 d = device.DeviceSpec(device_type="GPU", device_index=0)
48 d = device.DeviceSpec()
71 d = device.DeviceSpec(job="foo", replica=12, task=3, device_type="GPU")
75 d = device.DeviceSpec()
89 d = device.DeviceSpec.from_string("/job:foo/replica:0")
92 d = device.DeviceSpec.from_string("/job:muu/device:GPU:2/cpu:0")
95 d = device.DeviceSpec.from_string("/job:foo/replica:0/task:3/cpu:*")
97 d = device.DeviceSpec.from_string("/job:foo/replica:0/task:3/gpu:7"
    [all...]
device.py 25 @tf_export("DeviceSpec")
26 class DeviceSpec(object):
29 `DeviceSpec`s are used throughout TensorFlow to describe where state is stored
30 and computations occur. Using `DeviceSpec` allows you to parse device spec
37 device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
44 If a `DeviceSpec` is partially specified, it will be merged with other
45 `DeviceSpec`s according to the scope in which it is defined. `DeviceSpec`
50 with tf.device(DeviceSpec(job="train", )):
51 with tf.device(DeviceSpec(job="ps", device_type="GPU", device_index=0)
    [all...]
framework_lib.py 75 from tensorflow.python.framework.device import DeviceSpec
importer.py 335 coloc_device = pydev.DeviceSpec.from_string(coloc_op.device)
753 coloc_device = pydev.DeviceSpec.from_string(coloc_op.device)
ops_test.py 183 nodedef = ops._NodeDef("foo", "bar", device=pydev.DeviceSpec(job="j"))
269 pydev.DeviceSpec(
    [all...]
ops.py     [all...]
  /external/tensorflow/tensorflow/python/training/
device_setter.py 103 current_device = pydev.DeviceSpec.from_string(op.device or "")
111 ps_device = pydev.DeviceSpec.from_string(self._ps_device)
120 worker_device = pydev.DeviceSpec.from_string(self._worker_device or "")
193 ps_job_name = pydev.DeviceSpec.from_string(ps_device).job
saver.py 78 parsed_device = pydev.DeviceSpec.from_string(device_string)
    [all...]
  /external/tensorflow/tensorflow/python/eager/
context.py 63 self.device_spec = pydev.DeviceSpec.from_string(
314 device_spec = pydev.DeviceSpec.from_string(name)
318 new_device_spec = pydev.DeviceSpec.from_string(
322 new_device_spec = pydev.DeviceSpec.from_string("")
  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
replicate_model_fn.py 586 current_device = framework_device.DeviceSpec.from_string(op.device or '')
590 ps_device_spec = framework_device.DeviceSpec.from_string(
596 worker_device_spec = framework_device.DeviceSpec.from_string(
  /external/tensorflow/tensorflow/contrib/framework/python/ops/
variables.py 734 device_spec = tf_device.DeviceSpec(device_type=self._device_type,

Completed in 8362 milliseconds