Searched refs:logical_devices (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | test_util.py | 89 logical_devices = []
92 logical_devices.append(
95 logical_devices.append(context.LogicalDeviceConfiguration())
98 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
|
| D | tpu_strategy_test.py | 1271 logical_devices = []
1279 logical_devices.append((y.device, z.device))
1286 logical_devices)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_system_metadata.py | 80 logical_devices = config.list_logical_devices()
85 for d in logical_devices]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 809 def set_logical_device_configuration(device, logical_devices): argument
874 context.context().set_logical_device_configuration(device, logical_devices)
|
| D | config_test.py | 511 logical_devices = config.list_logical_devices()
514 config.get_device_details(logical_devices[0])
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 45 logical_devices = config.list_logical_devices()
47 for d in logical_devices:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | functional_ops_test.py | 580 logical_devices = []
581 logical_devices.append(context.LogicalDeviceConfiguration())
582 logical_devices.append(context.LogicalDeviceConfiguration())
584 framework_config.list_physical_devices("CPU")[0], logical_devices)
1019 logical_devices = []
1020 logical_devices.append(context.LogicalDeviceConfiguration())
1021 logical_devices.append(context.LogicalDeviceConfiguration())
1023 framework_config.list_physical_devices("CPU")[0], logical_devices)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.pbtxt | 73 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.config.experimental.pbtxt | 89 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.pbtxt | 73 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.config.experimental.pbtxt | 89 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable_test.py | 73 logical_devices = []
75 logical_devices.append(context.LogicalDeviceConfiguration())
78 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 81 logical_devices = [
85 logical_devices)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 504 logical_devices = []
517 logical_devices.append(
524 self._logical_devices = logical_devices
|