mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
642 B
30 lines
642 B
3 years ago
|
//
|
||
|
// 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
|