Searched refs:LogicalDeviceConfiguration (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 368 context.LogicalDeviceConfiguration(),
369 context.LogicalDeviceConfiguration()
397 context.LogicalDeviceConfiguration(),
398 context.LogicalDeviceConfiguration(),
399 context.LogicalDeviceConfiguration()
404 context.LogicalDeviceConfiguration(),
405 context.LogicalDeviceConfiguration()
473 context.LogicalDeviceConfiguration(memory_limit=10),
474 context.LogicalDeviceConfiguration(memory_limit=10)
493 context.LogicalDeviceConfiguration(memory_limit=20),
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.-logical-device-configuration.pbtxt | 1 path: "tensorflow.config.LogicalDeviceConfiguration"
3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
| D | tensorflow.config.experimental.-virtual-device-configuration.pbtxt | 3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
| D | tensorflow.config.pbtxt | 8 name: "LogicalDeviceConfiguration"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.-logical-device-configuration.pbtxt | 1 path: "tensorflow.config.LogicalDeviceConfiguration"
3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
| D | tensorflow.config.experimental.-virtual-device-configuration.pbtxt | 3 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
4 is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
| D | tensorflow.config.pbtxt | 8 name: "LogicalDeviceConfiguration"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | layer_correctness_test.py | 62 context.LogicalDeviceConfiguration(),
63 context.LogicalDeviceConfiguration(),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_continuous_run_test.py | 75 gpus[0], [context.LogicalDeviceConfiguration(64)])
|
| D | strategy_combinations.py | 210 context.LogicalDeviceConfiguration() for _ in range(num_virtual_cpus)
|
| D | mirrored_variable_test.py | 56 context.LogicalDeviceConfiguration(),
57 context.LogicalDeviceConfiguration(),
|
| D | mirrored_strategy_test.py | 1368 [context.LogicalDeviceConfiguration()] * 2)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | multi_gpu_utils_test.py | 50 context.LogicalDeviceConfiguration(500),
51 context.LogicalDeviceConfiguration(500)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 59 context.LogicalDeviceConfiguration(),
60 context.LogicalDeviceConfiguration(),
61 context.LogicalDeviceConfiguration(),
62 context.LogicalDeviceConfiguration()
|
| D | context.py | 262 class LogicalDeviceConfiguration( class
279 return super(LogicalDeviceConfiguration, cls).__new__(cls, memory_limit)
1166 cpus[0], [LogicalDeviceConfiguration() for _ in range(num_cpus)])
|
| D | core_test.py | 73 context.LogicalDeviceConfiguration(),
74 context.LogicalDeviceConfiguration()
|
| D | function_test.py | 137 context.LogicalDeviceConfiguration(),
138 context.LogicalDeviceConfiguration(),
139 context.LogicalDeviceConfiguration(),
140 context.LogicalDeviceConfiguration()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 386 context.LogicalDeviceConfiguration(),
387 context.LogicalDeviceConfiguration()
|
| D | stateful_random_ops_test.py | 66 context.LogicalDeviceConfiguration(),
67 context.LogicalDeviceConfiguration()
|
| /external/tensorflow/ |
| D | RELEASE.md | 54 …experimentalVirtualDeviceConfiguration` has been renamed to `tf.config.LogicalDeviceConfiguration`.
|