Support TensorRT provider (#921)

Signed-off-by: manickavela1998@gmail.com <manickavela1998@gmail.com> Signed-off-by: manickavela1998@gmail.com <manickavela.arumugam@uniphore.com>
2024-06-06 08:15:28 +05:30
parent 7e0931c762
commit 69347ffc8f
3 changed files with 64 additions and 1 deletions
--- a/sherpa-onnx/csrc/provider.h
+++ b/sherpa-onnx/csrc/provider.h
@@ -18,6 +18,7 @@ enum class Provider {
  kCoreML = 2,   // CoreMLExecutionProvider
  kXnnpack = 3,  // XnnpackExecutionProvider
  kNNAPI = 4,    // NnapiExecutionProvider
+  kTRT = 5,     // TensorRTExecutionProvider
 };

 /**