diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-24 17:07:43 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-10-25 14:51:29 +0000 |
commit | c82c8732fb514b412012002bd951a84039eca696 (patch) | |
tree | 3f332d585fd86ff45ba514eb069be7fe26a2004a /tests/InferenceModel.hpp | |
parent | 1f826967da321f87b192bd1c29152b0f1fc51ed8 (diff) | |
download | armnn-c82c8732fb514b412012002bd951a84039eca696.tar.gz |
IVGCVSW-4008 Add profiling mode to ExecuteNetwork
* Removed the requirement for specifying a data file for each
input tensor
* Added the possibility to generate dummy tensor data (filled
with 0s) if no data files are specified by the user
* Warn the user when they request to save the output to a file,
but the input was generate, therefore rendering the output
useless
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I8baed116dcd99fe380e419db322dc7e04ab1c653
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 0ede8983cf..412476a4df 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -442,6 +442,12 @@ public: } } + unsigned int GetInputSize(unsigned int inputIndex = 0u) const + { + CheckInputIndexIsValid(inputIndex); + return m_InputBindings[inputIndex].second.GetNumElements(); + } + unsigned int GetOutputSize(unsigned int outputIndex = 0u) const { CheckOutputIndexIsValid(outputIndex); |