[Java] JNI refactor for OrtJniUtil

java/src/main/java/ai/onnxruntime/MapInfo.java

@@ -4,6 +4,8 @@
  */
 package ai.onnxruntime;
 
+import ai.onnxruntime.TensorInfo.OnnxTensorType;
+
 /** Describes an {@link OnnxMap} object or output node. */
 public class MapInfo implements ValueInfo {
 
@@ -42,6 +44,21 @@ public class MapInfo implements ValueInfo {
     this.valueType = valueType;
   }
 
+  /**
+   * Construct a MapInfo with the specified size, key type and value type.
+   *
+   * <p>Called from JNI.
+   *
+   * @param size The size.
+   * @param keyTypeInt The int representing the {@link OnnxTensorType} of the keys.
+   * @param valueTypeInt The int representing the {@link OnnxTensorType} of the values.
+   */
+  MapInfo(int size, int keyTypeInt, int valueTypeInt) {
+    this.size = size;
+    this.keyType = OnnxJavaType.mapFromOnnxTensorType(OnnxTensorType.mapFromInt(keyTypeInt));
+    this.valueType = OnnxJavaType.mapFromOnnxTensorType(OnnxTensorType.mapFromInt(valueTypeInt));
+  }
+
   @Override
   public String toString() {
     String initial = size == -1 ? "MapInfo(size=UNKNOWN" : "MapInfo(size=" + size;

java/src/main/java/ai/onnxruntime/SequenceInfo.java

@@ -4,6 +4,8 @@
  */
 package ai.onnxruntime;
 
+import ai.onnxruntime.TensorInfo.OnnxTensorType;
+
 /** Describes an {@link OnnxSequence}, including it's element type if known. */
 public class SequenceInfo implements ValueInfo {
 
@@ -35,6 +37,23 @@ public class SequenceInfo implements ValueInfo {
     this.mapInfo = null;
   }
 
+  /**
+   * Construct a sequence of known length, with the specified type. This sequence does not contain
+   * maps.
+   *
+   * <p>Called from JNI.
+   *
+   * @param length The length of the sequence.
+   * @param sequenceTypeInt The element type int of the sequence mapped from {@link OnnxTensorType}.
+   */
+  SequenceInfo(int length, int sequenceTypeInt) {
+    this.length = length;
+    this.sequenceType =
+        OnnxJavaType.mapFromOnnxTensorType(OnnxTensorType.mapFromInt(sequenceTypeInt));
+    this.sequenceOfMaps = false;
+    this.mapInfo = null;
+  }
+
   /**
    * Construct a sequence of known length containing maps.
    *

java/src/main/java/ai/onnxruntime/TensorInfo.java

@@ -120,6 +120,20 @@ public static OnnxTensorType mapFromJavaType(OnnxJavaType type) {
     this.onnxType = onnxType;
   }
 
+  /**
+   * Constructs a TensorInfo with the specified shape and native type int.
+   *
+   * <p>Called from JNI.
+   *
+   * @param shape The tensor shape.
+   * @param typeInt The native type int.
+   */
+  TensorInfo(long[] shape, int typeInt) {
+    this.shape = shape;
+    this.onnxType = OnnxTensorType.mapFromInt(typeInt);
+    this.type = OnnxJavaType.mapFromOnnxTensorType(this.onnxType);
+  }
+
   /**
    * Get a copy of the tensor's shape.
    *