configureDevicesForMultiDeviceTest (reference) in projects: aosp_15_r20

Searched refs:configureDevicesForMultiDeviceTest (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A D	multi_device_iterator_test.py	`52 self._devices = self.configureDevicesForMultiDeviceTest(3) 132 self._devices = self.configureDevicesForMultiDeviceTest(3) 324 self._devices = self.configureDevicesForMultiDeviceTest(3)`
H A D	memory_cleanup_test.py	`43 self._devices = self.configureDevicesForMultiDeviceTest(3)`
H A D	test_base.py	`420 def configureDevicesForMultiDeviceTest(self, num_devices): member in DatasetTestBase`
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
H A D	multi_device_test.py	`34 self._devices = self.configureDevicesForMultiDeviceTest(2)`
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A D	prefetch_with_slack_test.py	`32 self._devices = self.configureDevicesForMultiDeviceTest(3)`