Fix

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,14 +631,17 @@ public String toString() {
     }
   }
 
-  static class Tensor_unknown extends Tensor {
+  static class Tensor_unsupported extends Tensor {
     private final ByteBuffer data;
     private final DType myDtype;
 
-    private Tensor_unknown(ByteBuffer data, long[] shape, DType dtype) {
+    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
@@ -647,7 +651,8 @@ public DType dtype() {
 
     @Override
     public String toString() {
-      return String.format("Tensor(%s, dtype=%d)", Arrays.toString(shape), this.myDtype);
+      return String.format(
+          "Unsupported tensor(%s, dtype=%d)", Arrays.toString(shape), this.myDtype);
     }
   }
 
@@ -696,7 +701,7 @@ private static Tensor nativeNewTensor(
     } else if (DType.INT8.jniCode == dtype) {
       tensor = new Tensor_int8(data, shape);
     } else {
-      tensor = new Tensor_unknown(data, shape, dtype);
+      tensor = new Tensor_unsupported(data, shape, DType.fromJniCode(dtype));
     }
     tensor.mHybridData = hybridData;
     return tensor;