Searched refs:LogicalDevice (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 22 from tensorflow.python.eager.context import LogicalDevice
50 LogicalDevice("/job:worker/task:0/device:GPU:0", "GPU"),
51 LogicalDevice("/job:worker/task:0/device:GPU:1", "GPU"),
52 LogicalDevice("/job:worker/task:0/device:GPU:2", "GPU"),
53 LogicalDevice("/job:worker/task:0/device:GPU:3", "GPU"),
70 LogicalDevice("/job:worker/task:0/device:TPU:0", "TPU"),
71 LogicalDevice("/job:worker/task:0/device:TPU:1", "TPU"),
72 LogicalDevice("/job:worker/task:0/device:TPU:2", "TPU"),
73 LogicalDevice("/job:worker/task:0/device:TPU:3", "TPU"),
74 LogicalDevice("/job:worker/task:0/device:GPU:0", "GPU"),
[all …]
|
| D | tfconfig_cluster_resolver_test.py | 22 from tensorflow.python.eager.context import LogicalDevice
300 LogicalDevice('/job:worker1/task:0/device:TPU:0', 'TPU'),
301 LogicalDevice('/job:worker1/task:0/device:TPU:1', 'TPU'),
302 LogicalDevice('/job:worker1/task:0/device:GPU:0', 'GPU'),
303 LogicalDevice('/job:worker1/task:0/device:GPU:1', 'GPU'),
304 LogicalDevice('/job:worker2/task:1/device:TPU:2', 'TPU'),
305 LogicalDevice('/job:worker2/task:2/device:TPU:3', 'TPU'),
306 LogicalDevice('/job:worker2/task:3/device:GPU:2', 'GPU'),
307 LogicalDevice('/job:worker2/task:4/device:GPU:3', 'GPU'),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.-logical-device.pbtxt | 1 path: "tensorflow.config.LogicalDevice"
3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDevice\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDevice\'>"
|
| D | tensorflow.config.pbtxt | 4 name: "LogicalDevice"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.-logical-device.pbtxt | 1 path: "tensorflow.config.LogicalDevice"
3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDevice\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDevice\'>"
|
| D | tensorflow.config.pbtxt | 4 name: "LogicalDevice"
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver_test.py | 26 from tensorflow.python.eager.context import LogicalDevice
616 LogicalDevice('/job:tpu_worker/task:0/device:TPU:0', 'TPU'),
617 LogicalDevice('/job:tpu_worker/task:1/device:TPU:1', 'TPU'),
618 LogicalDevice('/job:tpu_worker/task:2/device:TPU:0', 'TPU'),
619 LogicalDevice('/job:tpu_worker/task:3/device:TPU:1', 'TPU'),
620 LogicalDevice('/job:tpu_worker/task:0/device:TPU:4', 'TPU'),
621 LogicalDevice('/job:tpu_worker/task:1/device:TPU:5', 'TPU'),
622 LogicalDevice('/job:tpu_worker/task:2/device:TPU:4', 'TPU'),
623 LogicalDevice('/job:tpu_worker/task:3/device:TPU:5', 'TPU'),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | device_util.py | 46 if isinstance(d, context.LogicalDevice):
|
| D | mirrored_strategy.py | 70 name = d.name if isinstance(d, context.LogicalDevice) else d
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context_test.py | 182 dev = context.LogicalDevice(name='/job:worker/replica:0/task:1',
|
| D | context.py | 276 class LogicalDevice( class
547 LogicalDevice(name=spec.to_string(), device_type=spec.device_type))
999 if isinstance(name, LogicalDevice):
|
| /external/tensorflow/ |
| D | RELEASE.md | 6240 `tf.config.LogicalDevice`
|