LstmCommon.hpp

Go to the documentation of this file.

//
// 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>& stateH,
              const std::vector<float>& stateC)
            : m_InputSeq(inputSeq)
            , m_StateH(stateH)
            , m_StateC(stateC)
    {}

    std::vector<float>        m_InputSeq;
    std::vector<float>        m_StateH;
    std::vector<float>        m_StateC;
};

using LstmInputs = std::pair<std::string, std::vector<LstmInput>>;

} // anonymous namespace