diff options
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); |