xref: /aosp_15_r20/external/armnn/tests/LstmCommon.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <string>
8 #include <utility>
9 
10 namespace
11 {
12 
13 struct LstmInput
14 {
LstmInput__anon857632d30111::LstmInput15     LstmInput(const std::vector<float>& inputSeq,
16               const std::vector<float>& stateH,
17               const std::vector<float>& stateC)
18             : m_InputSeq(inputSeq)
19             , m_StateH(stateH)
20             , m_StateC(stateC)
21     {}
22 
23     std::vector<float>        m_InputSeq;
24     std::vector<float>        m_StateH;
25     std::vector<float>        m_StateC;
26 };
27 
28 using LstmInputs = std::pair<std::string, std::vector<LstmInput>>;
29 
30 } // anonymous namespace