OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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