tests/LstmCommon.hpp

   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 {
  15     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