Removing training-only nodes
The Arm NN SDK is an inference-only engine. For performance and compatibility, you must remove training-only nodes when using the TensorFlow neural network framework. These nodes include operations that you do not require for inference.
TensorFlow provides documentation on how to remove these training-only nodes.
You do not need to remove training-only nodes when you use the Caffe neural network framework as the Arm NN SDK handles the removal.