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
|