Set the list of visible devices.
(self, devices, device_type=None)
| 1162 | ] |
| 1163 | |
| 1164 | def set_visible_devices(self, devices, device_type=None): |
| 1165 | """Set the list of visible devices.""" |
| 1166 | self._initialize_physical_devices() |
| 1167 | |
| 1168 | if not isinstance(devices, list): |
| 1169 | devices = [devices] |
| 1170 | |
| 1171 | for d in devices: |
| 1172 | if d not in self._physical_devices: |
| 1173 | raise ValueError("Unrecognized device: %s" % repr(d)) |
| 1174 | if device_type is not None and d.device_type != device_type: |
| 1175 | raise ValueError("Unrecognized device: %s" % repr(d)) |
| 1176 | |
| 1177 | visible_device_list = [] |
| 1178 | if device_type is not None: |
| 1179 | visible_device_list = [ |
| 1180 | d for d in self._visible_device_list if d.device_type != device_type |
| 1181 | ] |
| 1182 | |
| 1183 | visible_device_list += devices |
| 1184 | |
| 1185 | if self._visible_device_list == visible_device_list: |
| 1186 | return |
| 1187 | |
| 1188 | if self._context_handle is not None: |
| 1189 | raise RuntimeError( |
| 1190 | "Visible devices cannot be modified after being initialized") |
| 1191 | |
| 1192 | self._visible_device_list = visible_device_list |
| 1193 | |
| 1194 | def get_memory_growth(self, dev): |
| 1195 | """Get if memory growth is enabled for a PhysicalDevice.""" |