Searched refs:mock_list_devices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 50 def testNumAcceleratorsSuccess(self, mock_list_devices, argument
63 mock_list_devices.return_value = device_list
70 def testNumAcceleratorsMultiDeviceSuccess(self, mock_list_devices, argument
87 mock_list_devices.return_value = device_list
94 def testNumAcceleratorsFilterTasks(self, mock_list_devices, argument
111 mock_list_devices.return_value = device_list
|
| D | tpu_cluster_resolver_test.py | 627 def testNumAcceleratorsSuccess(self, mock_list_devices, argument
644 mock_list_devices.return_value = device_list
654 def testNumAcceleratorsRetryFailure(self, mock_list_devices, argument
657 mock_list_devices.side_effect = errors.DeadlineExceededError(
|
| D | tfconfig_cluster_resolver_test.py | 248 def testNumAcceleratorsFilterTasksByEnvVar(self, mock_list_devices, argument
279 mock_list_devices.return_value = device_list
|