Searched refs:LogicalDevice (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 26 from tensorflow.python.eager.context import LogicalDevice
54 LogicalDevice("/job:worker/task:0/device:GPU:0", "GPU"),
55 LogicalDevice("/job:worker/task:0/device:GPU:1", "GPU"),
56 LogicalDevice("/job:worker/task:0/device:GPU:2", "GPU"),
57 LogicalDevice("/job:worker/task:0/device:GPU:3", "GPU"),
74 LogicalDevice("/job:worker/task:0/device:TPU:0", "TPU"),
75 LogicalDevice("/job:worker/task:0/device:TPU:1", "TPU"),
76 LogicalDevice("/job:worker/task:0/device:TPU:2", "TPU"),
77 LogicalDevice("/job:worker/task:0/device:TPU:3", "TPU"),
78 LogicalDevice("/job:worker/task:0/device:GPU:0", "GPU"),
[all …]
|
| D | tfconfig_cluster_resolver_test.py | 26 from tensorflow.python.eager.context import LogicalDevice
281 LogicalDevice('/job:worker1/task:0/device:TPU:0', 'TPU'),
282 LogicalDevice('/job:worker1/task:0/device:TPU:1', 'TPU'),
283 LogicalDevice('/job:worker1/task:0/device:GPU:0', 'GPU'),
284 LogicalDevice('/job:worker1/task:0/device:GPU:1', 'GPU'),
285 LogicalDevice('/job:worker2/task:1/device:TPU:2', 'TPU'),
286 LogicalDevice('/job:worker2/task:2/device:TPU:3', 'TPU'),
287 LogicalDevice('/job:worker2/task:3/device:GPU:2', 'GPU'),
288 LogicalDevice('/job:worker2/task:4/device:GPU:3', 'GPU'),
|
| /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/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/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver_test.py | 29 from tensorflow.python.eager.context import LogicalDevice
618 LogicalDevice('/job:tpu_worker/task:0/device:TPU:0', 'TPU'),
619 LogicalDevice('/job:tpu_worker/task:1/device:TPU:1', 'TPU'),
620 LogicalDevice('/job:tpu_worker/task:2/device:TPU:0', 'TPU'),
621 LogicalDevice('/job:tpu_worker/task:3/device:TPU:1', 'TPU'),
622 LogicalDevice('/job:tpu_worker/task:0/device:TPU:4', 'TPU'),
623 LogicalDevice('/job:tpu_worker/task:1/device:TPU:5', 'TPU'),
624 LogicalDevice('/job:tpu_worker/task:2/device:TPU:4', 'TPU'),
625 LogicalDevice('/job:tpu_worker/task:3/device:TPU:5', 'TPU'),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | device_util.py | 48 if isinstance(d, context.LogicalDevice):
|
| D | mirrored_strategy.py | 69 name = d.name if isinstance(d, context.LogicalDevice) else d
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 242 class LogicalDevice( class
493 LogicalDevice(name=spec.to_string(), device_type=spec.device_type))
874 if isinstance(name, LogicalDevice):
|
| /external/tensorflow/ |
| D | RELEASE.md | 1624 * `tf.device` & `MirroredStrategy` now supports passing in a `tf.config.LogicalDevice`
|