Pre-Quantized Deep Learning Models Codified in ONNX to Enable Hardware/Software Co-Design
Abstract
This paper presents a methodology to separate the quantization process from the hardware-specific model compilation stage via a pre-quantized deep learning model description in standard ONNX format. Separating the quantization process from the model compilation stage enables independent development. The methodology is expressive to convey hardware-specific operations and to embed key quantization parameters into a ONNX model which enables hardware/software co-design. Detailed examples are given for both MLP and CNN based networks, which can be extended to other networks in a straightforward fashion.
- Publication:
-
arXiv e-prints
- Pub Date:
- October 2021
- DOI:
- 10.48550/arXiv.2110.01730
- arXiv:
- arXiv:2110.01730
- Bibcode:
- 2021arXiv211001730H
- Keywords:
-
- Computer Science - Machine Learning;
- Computer Science - Hardware Architecture
- E-Print:
- 5 pages, 6 figures