Mobile GPU support is yet a prototype feature available on the nightly build.
Prototype features are not documented.
You can find an example here:

  
      github.com
  
  
    pytorch/tutorials/blob/master/prototype_source/nnapi_mobilenetv2.rst 70
(Prototype) Convert MobileNetV2 to NNAPI
========================================

Introduction
------------

This tutorial shows how to prepare a computer vision model to use
`Android's Neural Networks API (NNAPI) <https://developer.android.com/ndk/guides/neuralnetworks>`_.
NNAPI provides access to powerful and efficient computational cores
on many modern Android devices.

PyTorch's NNAPI is currently in the "prototype" phase and only supports
a limited range of operators, but we expect to solidify the integration
and expand our operator support over time.


Environment
-----------

Install PyTorch and torchvision.


  This file has been truncated. show original