Merge pull request #74 from saddam213/TensorRT

Add TensorRT ExecutionProvider

Author: saddam213

OnnxStack.Core/Config/ExecutionProvider.cs

@@ -6,6 +6,7 @@ public enum ExecutionProvider
         Cuda = 1,
         Cpu = 2,
         CoreML = 3,
-        OpenVino = 4
+        OpenVino = 4,
+        TensorRT = 5
     }
 }

OnnxStack.Core/Extensions/Extensions.cs

@@ -51,6 +51,9 @@ public static SessionOptions GetSessionOptions(this OnnxModelConfig configuratio
                     };
                     sessionOptions.AppendExecutionProvider_OpenVINO(deviceId);
                     return sessionOptions;
+                case ExecutionProvider.TensorRT:
+                    sessionOptions.AppendExecutionProvider_Tensorrt(configuration.DeviceId.Value);
+                    return sessionOptions;
             }
         }