| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | config_test.py | 390 context.LogicalDeviceConfiguration(),
391 context.LogicalDeviceConfiguration()
422 context.LogicalDeviceConfiguration(),
423 context.LogicalDeviceConfiguration(),
424 context.LogicalDeviceConfiguration()
429 context.LogicalDeviceConfiguration(),
430 context.LogicalDeviceConfiguration()
536 context.LogicalDeviceConfiguration(memory_limit=10),
537 context.LogicalDeviceConfiguration(memory_limit=10)
557 context.LogicalDeviceConfiguration(memory_limit=20),
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A 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\'>"
|
| H A 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\'>"
|
| H A D | tensorflow.config.pbtxt | 8 name: "LogicalDeviceConfiguration"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A 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\'>"
|
| H A 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\'>"
|
| H A D | tensorflow.config.pbtxt | 8 name: "LogicalDeviceConfiguration"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | functional_saver_test.py | 45 context.LogicalDeviceConfiguration(),
46 context.LogicalDeviceConfiguration(),
47 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | packed_distributed_variable_test.py | 35 context.LogicalDeviceConfiguration(),
36 context.LogicalDeviceConfiguration(),
|
| H A D | test_util.py | 114 context.LogicalDeviceConfiguration(memory_limit=2048))
116 logical_devices.append(context.LogicalDeviceConfiguration())
|
| H A D | multi_worker_continuous_run_test.py | 64 self._gpus[0], [context.LogicalDeviceConfiguration(64)])
|
| H A D | mirrored_variable_test.py | 56 context.LogicalDeviceConfiguration(),
57 context.LogicalDeviceConfiguration(),
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | collective_ops_test.py | 449 context.LogicalDeviceConfiguration(),
450 context.LogicalDeviceConfiguration()
495 context.LogicalDeviceConfiguration(),
496 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | replicate_test.py | 47 context.LogicalDeviceConfiguration(),
48 context.LogicalDeviceConfiguration(),
49 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| H A D | parallel_device_test.py | 105 context.LogicalDeviceConfiguration(memory_limit=100),
106 context.LogicalDeviceConfiguration(memory_limit=100),
112 context.LogicalDeviceConfiguration(),
113 context.LogicalDeviceConfiguration(),
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| H A 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",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | device_placement_test.py | 154 context.LogicalDeviceConfiguration(),
155 context.LogicalDeviceConfiguration()
|
| H A D | function_gradients_test.py | 56 context.LogicalDeviceConfiguration(),
57 context.LogicalDeviceConfiguration(),
58 context.LogicalDeviceConfiguration(),
59 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | save_test.py | 525 context.LogicalDeviceConfiguration(),
526 context.LogicalDeviceConfiguration()
596 context.LogicalDeviceConfiguration(),
597 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/docs/ |
| H A D | tf_doctest.py | 144 tf.config.LogicalDeviceConfiguration(memory_limit=256)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | functional_ops_test.py | 577 logical_devices.append(context.LogicalDeviceConfiguration())
578 logical_devices.append(context.LogicalDeviceConfiguration())
1016 logical_devices.append(context.LogicalDeviceConfiguration())
1017 logical_devices.append(context.LogicalDeviceConfiguration())
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| H A D | run_models.py | 92 virtual_device_config = context.LogicalDeviceConfiguration(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/ |
| H A D | stateful_random_ops_test.py | 58 context.LogicalDeviceConfiguration(),
59 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| H A D | rpc_ops_test.py | 48 context.LogicalDeviceConfiguration(),
49 context.LogicalDeviceConfiguration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | test_base.py | 436 context.LogicalDeviceConfiguration() for _ in range(num_devices)
|