Home
last modified time | relevance | path

Searched refs:logical_devices (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtest_util.py81 logical_devices = []
84 logical_devices.append(
87 logical_devices.append(context.LogicalDeviceConfiguration())
90 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
Dtpu_strategy_test.py1122 logical_devices = []
1130 logical_devices.append((y.device, z.device))
1137 logical_devices)
/external/tensorflow/tensorflow/python/framework/
Dconfig.py754 def set_logical_device_configuration(device, logical_devices): argument
815 context.context().set_logical_device_configuration(device, logical_devices)
Dconfig_test.py513 logical_devices = config.list_logical_devices()
516 config.get_device_details(logical_devices[0])
/external/tensorflow/tensorflow/python/tpu/
Dtpu_system_metadata.py80 logical_devices = config.list_logical_devices()
85 for d in logical_devices]
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py45 logical_devices = config.list_logical_devices()
47 for d in logical_devices:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.config.pbtxt73 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.config.experimental.pbtxt85 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.config.pbtxt73 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.config.experimental.pbtxt85 argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_interop_test.py81 logical_devices = [
85 logical_devices)
/external/tensorflow/tensorflow/python/eager/
Dcontext.py479 logical_devices = []
492 logical_devices.append(
499 self._logical_devices = logical_devices