INNER CODE UNIT · Python

supports_device

onnx/onnx-tensorrt · onnx_tensorrt/backend.py:260

    def supports_device(cls, device_str):
        device = Device(device_str)
        return device.type == DeviceType.CUDA

prepare         = TensorRTBackend.prepare
run_node        = TensorRTBackend.run_node
run_model       = TensorRTBackend.run_model
supports_device = TensorRTBackend.supports_device

View source record →