/aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
H A D | config_test.py | 389 config.set_logical_device_configuration(cpus[0], [ 421 config.set_logical_device_configuration(cpus[0], [ 428 config.set_logical_device_configuration(cpus[0], [ 535 config.set_logical_device_configuration(gpus[-1], [ 556 config.set_logical_device_configuration(gpus[-1], [ 562 config.set_logical_device_configuration(gpus[-1], [ 569 config.set_logical_device_configuration(gpus[-1], [ 734 config.set_logical_device_configuration(gpus[-1], [ 740 config.set_logical_device_configuration(gpus[-1], [
|
H A D | config.py | 809 def set_logical_device_configuration(device, logical_devices): function 874 context.context().set_logical_device_configuration(device, logical_devices)
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.config.pbtxt | 72 name: "set_logical_device_configuration"
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | multi_worker_continuous_run_test.py | 63 config.set_logical_device_configuration(
|
H A D | packed_distributed_variable_test.py | 34 config.set_logical_device_configuration(cpus[0], [
|
H A D | test_util.py | 119 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
|
H A D | mirrored_variable_test.py | 55 config.set_logical_device_configuration(cpus[0], [
|
H A D | collective_all_reduce_strategy_test.py | 690 tf_config.set_logical_device_configuration(device, configs)
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/docs/ |
H A D | tf_doctest.py | 147 tf.config.set_logical_device_configuration(available_gpus[0], logical_gpus)
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | collective_ops_test.py | 448 config.set_logical_device_configuration(cpus[0], [ 494 config.set_logical_device_configuration(cpus[0], [
|
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
H A D | run_models.py | 95 framework_config.set_logical_device_configuration(gpu,
|
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
H A D | functional_saver_test.py | 44 config.set_logical_device_configuration(cpus[0], [
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
H A D | parallel_device_test.py | 104 ctx.set_logical_device_configuration(gpus[0], [ 111 ctx.set_logical_device_configuration(cpus[0], [
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
H A 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",
|
H A D | TensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb | 66 "tf.config.set_logical_device_configuration(\n",
|
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
H A D | device_placement_test.py | 153 context.context().set_logical_device_configuration(cpus[0], [
|
H A D | context.py | 1514 self.set_logical_device_configuration( 1646 def set_logical_device_configuration(self, dev, virtual_devices): member in Context
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
H A D | test_base.py | 435 config.set_logical_device_configuration(cpus[0], [
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
H A D | np_interop_test.py | 80 tf.config.set_logical_device_configuration(physical_devices[0],
|
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
H A D | save_test.py | 523 context.context().set_logical_device_configuration( 594 context.context().set_logical_device_configuration(
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
H A D | replicate_test.py | 46 config.set_logical_device_configuration(cpus[0], [
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
H A D | functional_ops_test.py | 579 framework_config.set_logical_device_configuration( 1018 framework_config.set_logical_device_configuration(
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/ |
H A D | stateful_random_ops_test.py | 56 config.set_logical_device_configuration(
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
H A D | rpc_ops_test.py | 47 config.set_logical_device_configuration(cpus[0], [
|