[Android] Add unknown Tensor type instead of crash

Author: kirklandsign

extension/android/executorch_android/src/main/java/org/pytorch/executorch/DType.java

@@ -73,4 +73,13 @@ public enum DType {
   DType(int jniCode) {
     this.jniCode = jniCode;
   }
+
+  public static DType fromJniCode(int jniCode) {
+    for (DType dtype : values()) {
+      if (dtype.jniCode == jniCode) {
+        return dtype;
+      }
+    }
+    throw new IllegalArgumentException("No DType found for jniCode " + jniCode);
+  }
 }

extension/android/executorch_android/src/main/java/org/pytorch/executorch/Tensor.java

@@ -8,6 +8,7 @@
 
 package org.pytorch.executorch;
 
+import android.util.Log;
 import com.facebook.jni.HybridData;
 import com.facebook.jni.annotations.DoNotStrip;
 import java.nio.Buffer;
@@ -630,6 +631,31 @@ public String toString() {
     }
   }
 
+  static class Tensor_unsupported extends Tensor {
+    private final ByteBuffer data;
+    private final DType myDtype;
+
+    private Tensor_unsupported(ByteBuffer data, long[] shape, DType dtype) {
+      super(shape);
+      this.data = data;
+      this.myDtype = dtype;
+      Log.e(
+          "ExecuTorch",
+          toString() + " in Java. Please consider re-export the model with proper return type");
+    }
+
+    @Override
+    public DType dtype() {
+      return myDtype;
+    }
+
+    @Override
+    public String toString() {
+      return String.format(
+          "Unsupported tensor(%s, dtype=%d)", Arrays.toString(shape), this.myDtype);
+    }
+  }
+
   // region checks
   private static void checkArgument(boolean expression, String errorMessage, Object... args) {
     if (!expression) {
@@ -675,7 +701,7 @@ private static Tensor nativeNewTensor(
     } else if (DType.INT8.jniCode == dtype) {
       tensor = new Tensor_int8(data, shape);
     } else {
-      throw new IllegalArgumentException("Unknown Tensor dtype");
+      tensor = new Tensor_unsupported(data, shape, DType.fromJniCode(dtype));
     }
     tensor.mHybridData = hybridData;
     return tensor;