Configuration class for virtual devices for a PhysicalDevice. Fields: memory_limit: (optional) Maximum memory (in MB) to allocate on the virtual device. Currently only supported for GPUs.
| 256 | |
| 257 | @tf_export("config.experimental.VirtualDeviceConfiguration") |
| 258 | class VirtualDeviceConfiguration( |
| 259 | collections.namedtuple("VirtualDeviceConfiguration", ["memory_limit"])): |
| 260 | """Configuration class for virtual devices for a PhysicalDevice. |
| 261 | |
| 262 | Fields: |
| 263 | memory_limit: (optional) Maximum memory (in MB) to allocate on the virtual |
| 264 | device. Currently only supported for GPUs. |
| 265 | """ |
| 266 | |
| 267 | def __new__(cls, memory_limit=None): |
| 268 | return super(VirtualDeviceConfiguration, cls).__new__(cls, memory_limit) |
| 269 | |
| 270 | |
| 271 | class PhysicalDevice( |