Open Mohit0928 opened 2 years ago
This is something I'd like to do as well.
I've been reluctant to until now because it means the compiler frontend crate will now add TensorFlow Lite (either tflite
or librunecoral
) as a dependency... But that's a bloody big dependency and really hurts compile times if you want to just use the compiler frontend (e.g. Forge).
To implement this, during the "lowering" phase we need to associate models with minetypes based on the filename's extension. Then during the "typecheck" phase we need to add a pass which looks for TensorFlow Lite models, loads them, and checks input/output types.
I'm typing the wrong input/output type and dimensions in the
Runefile.yml
, but even then, it's building arune
. @Michael-F-Bryan, It should throw an error if I write a wrong input/output type and dimension in the Runefile.yml. Below is an example:Model info:
Typing a wrong Runefile.yml: