Home
last modified time | relevance | path

Searched refs:set_logical_device_configuration (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py389 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 Dconfig.py809 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 Dtensorflow.config.pbtxt72 name: "set_logical_device_configuration"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.config.pbtxt72 name: "set_logical_device_configuration"
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dmulti_worker_continuous_run_test.py63 config.set_logical_device_configuration(
H A Dpacked_distributed_variable_test.py34 config.set_logical_device_configuration(cpus[0], [
H A Dtest_util.py119 config.set_logical_device_configuration(physical_devices[-1], logical_devices)
H A Dmirrored_variable_test.py55 config.set_logical_device_configuration(cpus[0], [
H A Dcollective_all_reduce_strategy_test.py690 tf_config.set_logical_device_configuration(device, configs)
/aosp_15_r20/external/tensorflow/tensorflow/tools/docs/
H A Dtf_doctest.py147 tf.config.set_logical_device_configuration(available_gpus[0], logical_gpus)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcollective_ops_test.py448 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 Drun_models.py95 framework_config.set_logical_device_configuration(gpu,
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Dfunctional_saver_test.py44 config.set_logical_device_configuration(cpus[0], [
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/
H A Dparallel_device_test.py104 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 DTensorFlow_Numpy_Distributed_Image_Classification.ipynb98 " tf.config.set_logical_device_configuration(gpus[0], [\n",
104 " tf.config.set_logical_device_configuration(cpus[0], [\n",
H A DTensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb66 "tf.config.set_logical_device_configuration(\n",
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddevice_placement_test.py153 context.context().set_logical_device_configuration(cpus[0], [
H A Dcontext.py1514 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 Dtest_base.py435 config.set_logical_device_configuration(cpus[0], [
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_interop_test.py80 tf.config.set_logical_device_configuration(physical_devices[0],
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dsave_test.py523 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 Dreplicate_test.py46 config.set_logical_device_configuration(cpus[0], [
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dfunctional_ops_test.py579 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 Dstateful_random_ops_test.py56 config.set_logical_device_configuration(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/
H A Drpc_ops_test.py47 config.set_logical_device_configuration(cpus[0], [

12