Home
last modified time | relevance | path

Searched refs:get_logical_device_configuration (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig.py768 def get_logical_device_configuration(device): function
802 return context.context().get_logical_device_configuration(device)
H A Dconfig_test.py534 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))
539 self.assertEqual(len(config.get_logical_device_configuration(gpus[-1])), 2)
739 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.config.pbtxt48 name: "get_logical_device_configuration"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.config.pbtxt48 name: "get_logical_device_configuration"
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtest_util_test.py79 self.assertLen(config.get_logical_device_configuration(cpu_device), 3)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_interop_test.py75 configs = tf.config.get_logical_device_configuration(physical_devices[0])
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1637 def get_logical_device_configuration(self, dev): member in Context
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md6148 `tf.config.get_logical_device_configuration`,