mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-01 07:42:18 +00:00
30 lines
642 B
C++
30 lines
642 B
C++
//
|
|
// 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
|