Searched refs:LogicalDeviceConfiguration (Results 1 – 25 of 33) sorted by relevance
12
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 390 context.LogicalDeviceConfiguration(),
391 context.LogicalDeviceConfiguration()
419 context.LogicalDeviceConfiguration(),
420 context.LogicalDeviceConfiguration(),
421 context.LogicalDeviceConfiguration()
426 context.LogicalDeviceConfiguration(),
427 context.LogicalDeviceConfiguration()
533 context.LogicalDeviceConfiguration(memory_limit=10),
534 context.LogicalDeviceConfiguration(memory_limit=10)
554 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/distribute/ |
| D | mirrored_variable_test.py | 38 context.LogicalDeviceConfiguration(),
39 context.LogicalDeviceConfiguration(),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | packed_distributed_variable_test.py | 38 context.LogicalDeviceConfiguration(),
39 context.LogicalDeviceConfiguration(),
|
| D | test_util.py | 85 context.LogicalDeviceConfiguration(memory_limit=2048))
87 logical_devices.append(context.LogicalDeviceConfiguration())
|
| D | multi_worker_continuous_run_test.py | 68 self._gpus[0], [context.LogicalDeviceConfiguration(64)])
|
| D | mirrored_variable_test.py | 55 context.LogicalDeviceConfiguration(),
56 context.LogicalDeviceConfiguration(),
|
| D | collective_all_reduce_strategy_test.py | 538 context.LogicalDeviceConfiguration(64),
539 context.LogicalDeviceConfiguration(64),
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver_test.py | 49 context.LogicalDeviceConfiguration(),
50 context.LogicalDeviceConfiguration(),
51 context.LogicalDeviceConfiguration()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 453 context.LogicalDeviceConfiguration(),
454 context.LogicalDeviceConfiguration()
485 context.LogicalDeviceConfiguration(),
486 context.LogicalDeviceConfiguration()
|
| D | stateful_random_ops_test.py | 67 context.LogicalDeviceConfiguration(),
68 context.LogicalDeviceConfiguration()
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | layer_correctness_test.py | 62 context.LogicalDeviceConfiguration(),
63 context.LogicalDeviceConfiguration(),
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 106 context.LogicalDeviceConfiguration(memory_limit=100),
107 context.LogicalDeviceConfiguration(memory_limit=100),
113 context.LogicalDeviceConfiguration(),
114 context.LogicalDeviceConfiguration(),
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | multi_gpu_utils_test.py | 51 context.LogicalDeviceConfiguration(500),
52 context.LogicalDeviceConfiguration(500)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | replicate_test.py | 51 context.LogicalDeviceConfiguration(),
52 context.LogicalDeviceConfiguration(),
53 context.LogicalDeviceConfiguration()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 99 " tf.config.LogicalDeviceConfiguration(memory_limit=128),\n",
100 " tf.config.LogicalDeviceConfiguration(memory_limit=128)])\n",
105 " tf.config.LogicalDeviceConfiguration(),\n",
106 " tf.config.LogicalDeviceConfiguration()])\n",
|
| /external/tensorflow/tensorflow/tools/docs/ |
| D | tf_doctest.py | 158 tf.config.LogicalDeviceConfiguration(memory_limit=256)
|
| /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 | 261 class LogicalDeviceConfiguration( class
283 return super(LogicalDeviceConfiguration,
1364 cpus[0], [LogicalDeviceConfiguration() for _ in range(num_cpus)])
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 523 context.LogicalDeviceConfiguration(),
524 context.LogicalDeviceConfiguration()
594 context.LogicalDeviceConfiguration(),
595 context.LogicalDeviceConfiguration()
|
12