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.94k stars 826 forks source link

feat(compression): add YAML format for specifying compression #2795

Closed rkuester closed 6 days ago

rkuester commented 6 days ago

Add a module for reading YAML strings that specify how a model should be compressed. Put an example in spec.EXAMPLE_YAML_SPEC.

specfiles are written during model development to specify which tensors should be compressed, by what method, and according to what parameters. specfiles are read by the compression tool. They are not used by the TFLM interpreter.

Add a YAML parser to the python third-party package dependencies.

BUG=#2636

suleshahid commented 6 days ago

@Mergifyio refresh