diff options
| author | Jim Flynn <jim.flynn@arm.com> | 2019-04-15 14:34:17 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-04-16 10:58:52 +0000 |
| commit | e571d33a4550ab3bea6f42dda3ec88d5924e9c00 (patch) | |
| tree | b2dd99b11ca9c9eec79277d39674ada4c194f2f8 /tests/LstmCommon.hpp | |
| parent | aab6aff4aa282810cb535eeec65e59741f1f4f0e (diff) | |
| download | armnn-e571d33a4550ab3bea6f42dda3ec88d5924e9c00.tar.gz | |
IVGCVSW-2855 Create TfLite reference test for DeepSpeechV1
Change-Id: I4492a85c8337bf4ea0eb998c88b9cbfc932dc4e6
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'tests/LstmCommon.hpp')
| -rwxr-xr-x | tests/LstmCommon.hpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/LstmCommon.hpp b/tests/LstmCommon.hpp new file mode 100755 index 0000000000..31c4d041c1 --- /dev/null +++ b/tests/LstmCommon.hpp @@ -0,0 +1,30 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include <string> +#include <utility> + +namespace +{ + +struct LstmInput +{ + LstmInput(const std::vector<float>& inputSeq, + const std::vector<float>& stateC, + const std::vector<float>& stateH) + : m_InputSeq(inputSeq) + , m_StateC(stateC) + , m_StateH(stateH) + {} + + std::vector<float> m_InputSeq; + std::vector<float> m_StateC; + std::vector<float> m_StateH; +}; + +using LstmInputs = std::pair<std::string, std::vector<LstmInput>>; + +} // anonymous namespace
\ No newline at end of file |