Searched refs:supported_dev (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_test.py | 161 supported_dev = virtual_cluster.GetSupportedDevices(grappler_item)
162 self.assertEqual(supported_dev['add'], ['/CPU:0', '/GPU:0'])
163 self.assertEqual(supported_dev['Sum'], ['/CPU:0', '/GPU:0'])
164 self.assertEqual(supported_dev['range'], ['/CPU:0', '/GPU:0'])
167 supported_dev = real_cluster.GetSupportedDevices(grappler_item)
169 self.assertEqual(supported_dev['add'], [
173 self.assertEqual(supported_dev['Sum'], [
178 self.assertEqual(supported_dev['range'],
181 self.assertEqual(supported_dev['add'],
|
| D | cluster.i | 276 for (const auto& supported_dev : supported_device_types) { in TF_GetSupportedDevices() local
277 const string& node = supported_dev.first; in TF_GetSupportedDevices()
281 const std::set<string>& candidates = supported_dev.second; in TF_GetSupportedDevices()
286 feasible = supported_dev.second; in TF_GetSupportedDevices()
|