Searched defs: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
70 def testNumAcceleratorsMultiDeviceSuccess(self, mock_list_devices, argument
94 def testNumAcceleratorsFilterTasks(self, mock_list_devices, argument
|
| D | tfconfig_cluster_resolver_test.py | 248 def testNumAcceleratorsFilterTasksByEnvVar(self, mock_list_devices, argument
|
| D | tpu_cluster_resolver_test.py | 627 def testNumAcceleratorsSuccess(self, mock_list_devices, argument
654 def testNumAcceleratorsRetryFailure(self, mock_list_devices, argument
|