Support for DenseTensor<bool>

Author: saddam213

OnnxStack.Core/Extensions/OrtValueExtensions.cs

@@ -59,12 +59,31 @@ public static OrtValue ToOrtValue(this DenseTensor<long> tensor, OnnxNamedMetada
             return OrtValue.CreateTensorValueFromMemory(OrtMemoryInfo.DefaultInstance, tensor.Buffer, tensor.Dimensions.ToLong());
         }
 
+
+        /// <summary>
+        /// Converts DenseTensor<double> to OrtValue.
+        /// </summary>
+        /// <param name="tensor">The tensor.</param>
+        /// <param name="metadata">The metadata.</param>
+        /// <returns></returns>
         public static OrtValue ToOrtValue(this DenseTensor<double> tensor, OnnxNamedMetadata metadata)
         {
             return OrtValue.CreateTensorValueFromMemory(OrtMemoryInfo.DefaultInstance, tensor.Buffer, tensor.Dimensions.ToLong());
         }
 
 
+        /// <summary>
+        /// Converts DenseTensor<bool> to OrtValue.
+        /// </summary>
+        /// <param name="tensor">The tensor.</param>
+        /// <param name="metadata">The metadata.</param>
+        /// <returns></returns>
+        public static OrtValue ToOrtValue(this DenseTensor<bool> tensor, OnnxNamedMetadata metadata)
+        {
+            return OrtValue.CreateTensorValueFromMemory(OrtMemoryInfo.DefaultInstance, tensor.Buffer, tensor.Dimensions.ToLong());
+        }
+
+
         /// <summary>
         /// Creates and allocates the output tensors buffer.
         /// </summary>

OnnxStack.Core/Model/OnnxInferenceParameters.cs

@@ -35,6 +35,18 @@ public void AddInput(OrtValue value)
         }
 
 
+        /// <summary>
+        /// Adds the input parameter.
+        /// </summary>
+        /// <param name="index">The index.</param>
+        /// <param name="value">The value.</param>
+        public void AddInput(int index, OrtValue value)
+        {
+            var metadata = _metadata.Inputs[index];
+            _inputs.Add(metadata, value);
+        }
+
+
         /// <summary>
         /// Adds the input tensor.
         /// </summary>
@@ -89,6 +101,17 @@ public void AddInputTensor(DenseTensor<long> value)
         }
 
 
+        /// <summary>
+        /// Adds the input tensor.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        public void AddInputTensor(DenseTensor<bool> value)
+        {
+            var metaData = GetNextInputMetadata();
+            _inputs.Add(metaData, value.ToOrtValue(metaData));
+        }
+
+
         /// <summary>
         /// Adds an output parameter with known output size.
         /// </summary>