tensorplay.accelerator.get_device_capability

tensorplay.accelerator.get_device_capability(device=None) → dict[str, Any][source]

Capability map for an accelerator device.

The map carries a supported_dtypes set listing the data types that can be allocated on the device.