Searched refs:set_logical_device_configuration (Results 1 – 25 of 35) sorted by relevance
12
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 384 config.set_logical_device_configuration(cpus[0], [
416 config.set_logical_device_configuration(cpus[0], [
423 config.set_logical_device_configuration(cpus[0], [
530 config.set_logical_device_configuration(gpus[-1], [
551 config.set_logical_device_configuration(gpus[-1], [
557 config.set_logical_device_configuration(gpus[-1], [
564 config.set_logical_device_configuration(gpus[-1], [
729 config.set_logical_device_configuration(gpus[-1], [
735 config.set_logical_device_configuration(gpus[-1], [
|
| D | config.py | 809 def set_logical_device_configuration(device, logical_devices): function
874 context.context().set_logical_device_configuration(device, logical_devices)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | mirrored_variable_test.py | 39 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/tools/docs/ |
| D | tf_doctest.py | 148 tf.config.set_logical_device_configuration(available_gpus[0], logical_gpus)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | packed_distributed_variable_test.py | 38 config.set_logical_device_configuration(cpus[0], [
|
| D | multi_worker_continuous_run_test.py | 67 config.set_logical_device_configuration(
|
| D | test_util.py | 98 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
|
| D | collective_all_reduce_strategy_test.py | 543 tf_config.set_logical_device_configuration(gpus[-1], [
|
| D | mirrored_variable_test.py | 59 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | run_models.py | 95 framework_config.set_logical_device_configuration(gpu,
|
| /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], [
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | layer_correctness_test.py | 73 config_module.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | multi_gpu_utils_test.py | 47 config.set_logical_device_configuration(gpu_devices[0], [
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | device_placement_test.py | 157 context.context().set_logical_device_configuration(cpus[0], [
|
| D | context.py | 1428 self.set_logical_device_configuration(
1559 def set_logical_device_configuration(self, dev, virtual_devices): member in Context
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver_test.py | 48 config.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 107 ctx.set_logical_device_configuration(gpus[0], [
114 ctx.set_logical_device_configuration(cpus[0], [
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | test_base.py | 416 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 | 524 context.context().set_logical_device_configuration(
595 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], [
|
12