• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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__anon14be0f090111::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