ucb-bar / gemmini

Berkeley's Spatial Array Generator
Other
715 stars 145 forks source link

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

Open johnny-wang16 opened 1 year ago

johnny-wang16 commented 1 year ago

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.

fzi-peccia commented 10 months ago

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

The code can be found here:

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:

johnny-wang16 commented 10 months ago

Thanks for work! I'll take a look!