Generalize test models to be reused
Change-Id: I47705cfde2788fa74f8bbcd07100188e4690dbf2
diff --git a/scripts/run_platform.py b/scripts/run_platform.py
index a39de8b..0629fef 100755
--- a/scripts/run_platform.py
+++ b/scripts/run_platform.py
@@ -1,7 +1,7 @@
#!/usr/bin/env python3
#
-# Copyright (c) 2021 Arm Limited. All rights reserved.
+# Copyright (c) 2021-2022 Arm Limited.
#
# SPDX-License-Identifier: Apache-2.0
#
@@ -48,7 +48,7 @@
if (index > 1):
raise Exception("Illegal index value - Should be '0' or '1'")
(ta_parsed)[index] = value
-
+
return ta_parsed
def build_core_platform(output_folder, target, toolchain, memory_model, memory_arena, pmu,
@@ -183,7 +183,12 @@
res = run_cmd(vela_cmd)
optimized_model_path = output_folder/(network_path.stem + "_vela.tflite")
model_name = network_path.stem
- dump_c_header(optimized_model_path, output_folder/"model.h", "networkModelData", "network_model_sec", 16, extra_data=f"const char *modelName=\"{model_name}\";\n")
+ dump_c_header(optimized_model_path, output_folder/"model.h", "networkModelData", "network_model_sec", 16, extra_data=f"""
+#include <stddef.h>
+
+const size_t tensorArenaSize = 2000000;
+const char* modelName = \"{model_name}\";
+""")
def run_model(output_folder):
build_folder = output_folder/"model"/"build"