diff options
Diffstat (limited to 'scripts/py/templates/tflite.cc.template')
-rw-r--r-- | scripts/py/templates/tflite.cc.template | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/scripts/py/templates/tflite.cc.template b/scripts/py/templates/tflite.cc.template index 97bdec5..a06c14a 100644 --- a/scripts/py/templates/tflite.cc.template +++ b/scripts/py/templates/tflite.cc.template @@ -16,7 +16,11 @@ #} {{common_template_header}} -#include "Model.hpp" +#include "BufAttributes.hpp" + +#include <cstddef> +#include <cstdint> + {% for header in additional_headers %} #include "{{header}}" {% endfor %} |