supports_device

dragon.vm.tensorrt.backend.supports_device(device_str)

Query if given device is supported.

Parameters:
  • device_str (str) – The device descriptor.
Returns:

boolTrue if device is supported otherwise False.