Deploying a TensorFlow MNIST model on Arm NN

Overview Load and parse the MNIST test set Import graph Optimize and load onto compute device Run graph on device Deploy an application using Arm NN Next steps
Single page Download PDF

Import graph

Arm NN provides parsers for reading model files from neural network frameworks. There are typically two steps to do this:

  1. Load the model.
  2. Bind the input and output points of its graph.

Code to import the TensorFlow model

TensorFlow graphs in both text and binary ProtoBuf formats are supported. For more details on freezing TensorFlow graphs to include their weights, see this guide.

Tip: After this step, the code is common regardless of the framework that you started with as the INetwork and two BindingPointInfo objects provide everything that is needed.

Previous Next