fastmachinelearning / qonnx

QONNX: Arbitrary-Precision Quantized Neural Networks in ONNX
https://qonnx.readthedocs.io/
Apache License 2.0
124 stars 39 forks source link

Support `QConv2DBatchnorm` and `QDenseBatchnorm` layers #10

Open jmduarte opened 2 years ago

jmduarte commented 2 years ago

Details

Currently, QConv2DBatchnorm and QDenseBatchnorm are not supported for conversion in #7

New behavior

Support QConv2DBatchnorm and QDenseBatchnorm layers for QKeras to QONNX conversion

Motivation

This would allow MLPerf Tiny models, which use these QKeras layers, to be converted from

Parts of QONNX being affected

Mainly (Q)Keras conversion files src/qonnx/converters/qkeras/qlayers.py and src/qonnx/converters/keras.py

@rushtheboy @nhanvtran @julesmuhizi

LorenzoLagostina commented 1 year ago

Hi, any updates on this?