Use to use GPU if available on mobile


I have neural network that I trained on my PC and then jit compiled and run inferences on my Android phone. I can see from Android Studio debugger it is processing using the mobile’s CPU. How can I check and use GPU if available on android mobile?

Thanks in advance


Unfortunately, mobile GPU is a completely different language and architecture to everything else. So we don’t support it yet.
It is work in progress though and you can see all the PRs working it here:

1 Like