Bump ONNX-Scala & ONNX opset and IR versions; support Boolean outputs

Author: EmergentOrder

backends/src/main/scala/ORTOperatorBackend.scala

@@ -117,6 +117,13 @@ trait ORTOperatorBackend
           buff.get(x)
         }.toArray
       }
+      case ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL =>{
+        val point = value.GetTensorMutableDataBool.capacity(size)
+        val booleanPoint = new BooleanPointer(point.asByteBuffer) //C++ bool size is not defined, could cause problems on some platforms
+        (0 until booleanPoint.capacity().toInt).map { x =>
+          booleanPoint.get(x)
+        }.toArray
+      }
     }
     TensorFactory.getTensor(arr, shape)
   }

backends/src/main/scala/ORTOperatorBackend213.scala

@@ -116,6 +116,13 @@ trait ORTOperatorBackend
           buff.get(x)
         }.toArray
       }
+      case ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL =>{
+        val point = value.GetTensorMutableDataBool.capacity(size)
+        val booleanPoint = new BooleanPointer(point.asByteBuffer) //C++ bool size is not defined, could cause problems on some platforms
+        (0 until booleanPoint.capacity().toInt).map { x =>
+          booleanPoint.get(x)
+        }.toArray
+      }
     }
     TensorFactory.getTensor(arr, shape)
   }

build.sbt

@@ -10,7 +10,7 @@ scalaVersion := scala213Version
 lazy val commonSettings = Seq(
 //  scalaJSUseMainModuleInitializer := true, //Test only
   organization := "org.emergentorder.onnx",
-  version := "0.3.0",
+  version := "0.4.0",
   scalaVersion := scala213Version,
   resolvers += Resolver.mavenLocal,
   resolvers += "Sonatype OSS Snapshots" at "https://oss.sonatype.org/content/repositories/snapshots",

core/src/main/scala/ONNX213.scala

@@ -3637,7 +3637,7 @@ package object onnx {
       ](name, "Equal", allInputs, map))
     }
 
-    def Equal11[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
+    def Equal11[@sp T: Numeric: ClassTag, @sp T1: ClassTag](
         name: String,
         A: Option[Tensor[T]],
         B: Option[Tensor[T]]
@@ -5123,7 +5123,7 @@ package object onnx {
       ](name, "Greater", allInputs, map))
     }
 
-    def Greater9[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
+    def Greater9[@sp T: Numeric: ClassTag, @sp T1: ClassTag](
         name: String,
         A: Option[Tensor[T]],
         B: Option[Tensor[T]]
@@ -6188,7 +6188,7 @@ package object onnx {
       ](name, "Less", allInputs, map))
     }
 
-    def Less9[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
+    def Less9[@sp T: Numeric: ClassTag, @sp T1: ClassTag](
         name: String,
         A: Option[Tensor[T]],
         B: Option[Tensor[T]]

core/src/main/scala/ONNXHelper.scala

@@ -28,7 +28,7 @@ import java.io.File
 import scala.reflect.io.Streamable
 import org.bytedeco.javacpp._
 import org.bytedeco.onnx._
-import org.bytedeco.onnx.global.onnx.ParseProtoFromBytes
+import org.bytedeco.onnx.global.onnx.{ParseProtoFromBytes,check_model}
 
 class ONNXHelper(val byteArray: Array[Byte]) extends AutoCloseable {
 
@@ -45,7 +45,7 @@ class ONNXHelper(val byteArray: Array[Byte]) extends AutoCloseable {
       bytes,
       byteArray.length.toLong
     )
-//    bytes.close
+    check_model(r)
     r
   }
 

core/src/main/scala/OpToONNXBytesConverter.scala

@@ -168,10 +168,10 @@ trait OpToONNXBytesConverter extends AutoCloseable {
 
 //    origNode.close
     model.set_allocated_graph(graph)
-    model.set_ir_version(3)
+    model.set_ir_version(6)
 
     model.add_opset_import
-    model.opset_import(0).set_version(8)
+    model.opset_import(0).set_version(12)
 
     val outputValueInfo = graph.add_output
 

core/src/main/scala/OpToONNXBytesConverter213.scala

@@ -224,10 +224,10 @@ trait OpToONNXBytesConverter extends AutoCloseable {
 
     origNode.close
     model.set_allocated_graph(graph)
-    model.set_ir_version(3)
+    model.set_ir_version(6)
 
     model.add_opset_import
-    model.opset_import(0).set_version(8)
+    model.opset_import(0).set_version(12)
 
     val outputValueInfo = graph.add_output