supports_device

dragon.vm.tensorrt.onnx.supports_device(device_str)[source]

Query if the given device is supported.

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

boolTrue if device is supported otherwise False.