From 5508acb0c7803769b9bbda08762b232c13272bd9 Mon Sep 17 00:00:00 2001 From: Davide Grohmann Date: Mon, 8 Aug 2022 17:12:08 +0200 Subject: Generalize test models to be reused Change-Id: I47705cfde2788fa74f8bbcd07100188e4690dbf2 --- scripts/run_platform.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'scripts') 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 @@ def ta_parse_raw(ta_raw): 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 @@ def optimize_network(output_folder, network_path, accelerator_conf): 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 + +const size_t tensorArenaSize = 2000000; +const char* modelName = \"{model_name}\"; +""") def run_model(output_folder): build_folder = output_folder/"model"/"build" -- cgit v1.2.1