Tensor 3G Mali/Rio support - Samsung & Pixel

Hi, I want to run a PyTorch model accelerated on new Samsung and Pixel devices.
I see that there is a tutorial for XNNPACK (CPU) and one for Hexagon NPU. However if I understand it correctly Samsung and Pixel phones use something called “Tensor 3G” with Mali GPU and Rio TPU.
Is there currently any support for accelerated Tensor 3G processing?

Thanks!

1 Like