tensorflow / tflite-micro

Infrastructure to enable deployment of ML models to low-power resource-constrained embedded targets (including microcontrollers and digital signal processors).
Apache License 2.0
1.93k stars 827 forks source link

feat(compression): add module for defining test .tflite models #2798

Closed rkuester closed 3 days ago

rkuester commented 3 days ago

Add a module that provides tools for constructing .tflite flatbuffers from a Python dictionary representation of a model. This is useful for declaratively defining model flatbuffers for testing, which is faster and clearer than building up model flatbuffers programmatically.

The implementation should stay low-level and independent from any helpers in this project which make constructing model and flatbuffers easier, because this module is used to define tests for those helpers.

BUG=#2636