Searched refs:available_devices (Results 1 – 4 of 4) sorted by relevance
172 available_devices = _get_available_devices()173 available_devices = [174 _normalize_device_name(name) for name in available_devices177 if device not in available_devices:182 available_devices))
28 available_devices = [32 if '/gpu:%d' % (gpus - 1) not in available_devices:
117 available_devices = [device.name for device in gcluster.ListDevices()]121 self.assertTrue(not node.device or node.device in available_devices)134 available_devices = [device.name for device in gcluster.ListDevices()]136 self.assertTrue(not node.device or node.device in available_devices)
581 std::vector<DeviceAttributes> available_devices; in RunAndFetch() local582 TF_ASSERT_OK(session->ListDevices(&available_devices)) in RunAndFetch()587 absl::c_any_of(available_devices, [](const DeviceAttributes& device) { in RunAndFetch()