Android_deploy app crashes on tflite and tensorflow models

Hi, I’m trying to run android_deploy app on android devices.

I’v tried to deploy mobilenet_v2 model from three different frontends which are Mxnet, Tensorflow, and Tflite on android devices.

For Mxnet model, android_deploy app ran without any problems.

However, for Tensorflow and Tflite models, android_deploy app crashes.

Here is the adb logcat:

02-27 16:45:12.200 16108 16840 E AndroidRuntime: Process: org.apache.tvm.android.demo, PID: 16108

02-27 16:45:12.200 16108 16840 E AndroidRuntime: Caused by: org.apache.tvm.Base$TVMError: TVMError: Check failed: data->shape[j] == data_out->shape[j] (1001 vs. 1000) :

02-27 16:45:12.200 16108 16840 E AndroidRuntime: at org.apache.tvm.Base.checkCall(Base.java:173)

02-27 16:45:12.200 16108 16840 E AndroidRuntime: at org.apache.tvm.Function.invoke(Function.java:130)

02-27 16:45:12.200 16108 16840 E AndroidRuntime: at org.apache.tvm.android.demo.MainActivity$ModelRunAsyncTask.doInBackground(MainActivity.java:357)

02-27 16:45:12.200 16108 16840 E AndroidRuntime: at org.apache.tvm.android.demo.MainActivity$ModelRunAsyncTask.doInBackground(MainActivity.java:313)

02-27 16:45:12.204 1691 1708 W ActivityManager: Force finishing activity org.apache.tvm.android.demo/.MainActivity

02-27 16:45:12.816 16108 16108 E WindowManager: android.view.WindowLeaked: Activity org.apache.tvm.android.demo.MainActivity has leaked window DecorView@95e8dd[] that was originally added here

02-27 16:45:12.816 16108 16108 E WindowManager: at org.apache.tvm.android.demo.MainActivity$ModelRunAsyncTask.onPreExecute(MainActivity.java:433)

02-27 16:45:12.816 16108 16108 E WindowManager: at org.apache.tvm.android.demo.MainActivity.onActivityResult(MainActivity.java:562)

I think the bold message is the root cause of the app crash.

Note I’m currently using Dockerfile.demo_android in TVM v0.7.

Please let me know if you have any idea to solve the above problem.

Thanks!