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
// |
|
// 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
|