Searched refs:set_logical_device_configuration (Results 1 – 25 of 30) sorted by relevance
12
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 389 config.set_logical_device_configuration(cpus[0], [
418 config.set_logical_device_configuration(cpus[0], [
425 config.set_logical_device_configuration(cpus[0], [
532 config.set_logical_device_configuration(gpus[-1], [
553 config.set_logical_device_configuration(gpus[-1], [
559 config.set_logical_device_configuration(gpus[-1], [
566 config.set_logical_device_configuration(gpus[-1], [
682 config.set_logical_device_configuration(gpus[-1], [
688 config.set_logical_device_configuration(gpus[-1], [
|
| D | config.py | 754 def set_logical_device_configuration(device, logical_devices): function
815 context.context().set_logical_device_configuration(device, logical_devices)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | mirrored_variable_test.py | 37 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | packed_distributed_variable_test.py | 37 config.set_logical_device_configuration(cpus[0], [
|
| D | multi_worker_continuous_run_test.py | 67 config.set_logical_device_configuration(
|
| D | test_util.py | 90 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
|
| D | mirrored_variable_test.py | 54 config.set_logical_device_configuration(cpus[0], [
|
| D | collective_all_reduce_strategy_test.py | 537 tf_config.set_logical_device_configuration(gpus[-1], [
|
| /external/tensorflow/tensorflow/tools/docs/ |
| D | tf_doctest.py | 161 tf.config.set_logical_device_configuration(available_gpus[0], logical_gpus)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | layer_correctness_test.py | 61 config_module.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 452 config.set_logical_device_configuration(cpus[0], [
484 config.set_logical_device_configuration(cpus[0], [
|
| D | stateful_random_ops_test.py | 65 config.set_logical_device_configuration(
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 105 ctx.set_logical_device_configuration(gpus[0], [
112 ctx.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | multi_gpu_utils_test.py | 50 config.set_logical_device_configuration(gpu_devices[0], [
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver_test.py | 48 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 98 " tf.config.set_logical_device_configuration(gpus[0], [\n",
104 " tf.config.set_logical_device_configuration(cpus[0], [\n",
|
| D | TensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb | 66 "tf.config.set_logical_device_configuration(\n",
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 84 tf.config.set_logical_device_configuration(physical_devices[0],
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 521 context.context().set_logical_device_configuration(
592 context.context().set_logical_device_configuration(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | replicate_test.py | 50 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1363 self.set_logical_device_configuration(
1493 def set_logical_device_configuration(self, dev, virtual_devices): member in Context
|
| D | function_gradients_test.py | 58 config.set_logical_device_configuration(cpus[0], [
|
| D | core_test.py | 72 config.set_logical_device_configuration(cpus[0], [
|
12