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