diff options
Diffstat (limited to 'scripts/run_platform.py')
-rwxr-xr-x | scripts/run_platform.py | 11 |
1 files changed, 8 insertions, 3 deletions
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 <stddef.h> + +const size_t tensorArenaSize = 2000000; +const char* modelName = \"{model_name}\"; +""") def run_model(output_folder): build_folder = output_folder/"model"/"build" |