gemmini icon indicating copy to clipboard operation
gemmini copied to clipboard

[Feature Request] Support .tflite file and participate in the TinyML community

Open johnny-wang16 opened this issue 1 year ago • 2 comments

Hi, as Gemmini gains more and more attention, I was wondering if you guys have plans to support the .tflite format. By support i mean enabling the functionality of deploying a .tflite format model to run on Gemmini. kind of like supporting an onnx format model to run on Gemmini.

johnny-wang16 avatar Mar 17 '23 10:03 johnny-wang16

Hi! If you are interested in deploying TFLite to Gemmini, take a look at the integration I developed between TVM and Gemmini.

  • https://discuss.tvm.apache.org/t/presenting-the-generation-of-code-for-the-gemmini-accelerator/14107

The code can be found here:

  • https://github.com/fzi-peccia/tvm/tree/gemmini-microtvm

I also added a pull request here to add the Gemmini support for a non-zero padding, which is needed to deploy quantized TFLite models:

  • https://github.com/ucb-bar/gemmini/pull/274

fzi-peccia avatar Aug 29 '23 10:08 fzi-peccia

Thanks for work! I'll take a look!

johnny-wang16 avatar Aug 29 '23 12:08 johnny-wang16