Skip to content

CMSIS_NN integration

Wissam Boussella requested to merge cmsis_nn into master

This merge request contains tools for exporting quantized neural networks to Arm Cortex M targets, using the Arm CMSIS_NN library.

It contains :

  • New templates for generating code (Fully Connected, Conv, Max Pooling, Relu)
  • A merge of the Conv Relu Scaling and Fully Connected Relu Scaling nodes
  • A function to convert a scaling parameter into a shift and a multiply.
  • Unit tests in the form of small layer exports, with expected outputs in the console

This export is optimized for CMSIS_NN 6.0

For the moment, only 8-bit quantization is available; a 4-bit and 16-bit version will soon be available.

Note: several TODOs remain, including a Datatype problem not yet solved, an input problem after quantization, etc.

Merge request reports

Loading