diff --git a/src/backends/onnx.js b/src/backends/onnx.js index 38cd71337..df2c24be8 100644 --- a/src/backends/onnx.js +++ b/src/backends/onnx.js @@ -38,6 +38,7 @@ const DEVICE_TO_EXECUTION_PROVIDER_MAPPING = Object.freeze({ webgpu: 'webgpu', // WebGPU cuda: 'cuda', // CUDA dml: 'dml', // DirectML + tensorrt: 'tensorrt', // TensorRT webnn: { name: 'webnn', deviceType: 'cpu' }, // WebNN (default) 'webnn-npu': { name: 'webnn', deviceType: 'npu' }, // WebNN NPU diff --git a/src/utils/devices.js b/src/utils/devices.js index 1086b33e4..ebe5f6b4d 100644 --- a/src/utils/devices.js +++ b/src/utils/devices.js @@ -10,6 +10,7 @@ export const DEVICE_TYPES = Object.freeze({ webgpu: 'webgpu', // WebGPU cuda: 'cuda', // CUDA dml: 'dml', // DirectML + tensorrt: 'tensorrt', // TensorRT webnn: 'webnn', // WebNN (default) 'webnn-npu': 'webnn-npu', // WebNN NPU