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