diff options
author | Måns Nilsson <mans.nilsson@arm.com> | 2023-04-19 11:06:00 +0200 |
---|---|---|
committer | Måns Nilsson <mans.nilsson@arm.com> | 2023-04-26 10:25:17 +0200 |
commit | f21c31b5923a80fc4d74b54cf59dce3f0002d0f2 (patch) | |
tree | 34167c2fb232ff2473a96c33295bb5c9d82d4151 /applications | |
parent | c23c4e135b8982904b300feadbbd96fbf293398f (diff) | |
download | ethos-u-core-software-f21c31b5923a80fc4d74b54cf59dce3f0002d0f2.tar.gz |
Make inference process reusable
Private members are protected and run job is virtual to make it more
flexible to customize running jobs.
Change-Id: I2e4a0760000a451eead062ef758b467ea248dffe
Diffstat (limited to 'applications')
-rw-r--r-- | applications/inference_process/include/inference_process.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/applications/inference_process/include/inference_process.hpp b/applications/inference_process/include/inference_process.hpp index f8d7fd8..9bc0b56 100644 --- a/applications/inference_process/include/inference_process.hpp +++ b/applications/inference_process/include/inference_process.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2022 Arm Limited. All rights reserved. + * SPDX-FileCopyrightText: Copyright 2019-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> * * SPDX-License-Identifier: Apache-2.0 * @@ -75,9 +75,9 @@ class InferenceProcess { public: InferenceProcess(uint8_t *_tensorArena, size_t _tensorArenaSize); - bool runJob(InferenceJob &job); + virtual bool runJob(InferenceJob &job); -private: +protected: static bool copyIfm(InferenceJob &job, tflite::MicroInterpreter &interpreter); static bool copyOfm(InferenceJob &job, tflite::MicroInterpreter &interpreter); static bool compareOfm(InferenceJob &job, tflite::MicroInterpreter &interpreter); |