| 347 | } // namespace |
| 348 | |
| 349 | Status RegisterDeviceMapper(DeviceAllocationType device_type, DeviceMapper mapper) { |
| 350 | auto registry = GetDeviceRegistry(); |
| 351 | return registry->RegisterDevice(device_type, std::move(mapper)); |
| 352 | } |
| 353 | |
| 354 | Result<DeviceMapper> GetDeviceMapper(DeviceAllocationType device_type) { |
| 355 | auto registry = GetDeviceRegistry(); |