nmtSample::LSTMDecoder Class Reference

encodes single input into output states with LSTM More...

Inheritance diagram for nmtSample::LSTMDecoder:

Collaboration diagram for nmtSample::LSTMDecoder:

Public Types
typedef std::shared_ptr< Decoder >	ptr

Public Member Functions
	LSTMDecoder (ComponentWeights::ptr weights)
void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *inputEmbeddedData, nvinfer1::ITensor **inputStates, nvinfer1::ITensor **outputData, nvinfer1::ITensor **outputStates) override
	add the memory, cell, and hidden states to the network More...
std::vector< nvinfer1::Dims >	getStateSizes () override
	get the sizes (vector of them) of the hidden state vectors More...
std::string	getInfo () override
	get the textual description of the component More...
	~LSTMDecoder () override=default

Protected Attributes
ComponentWeights::ptr	mWeights
std::vector< nvinfer1::Weights >	mGateKernelWeights
std::vector< nvinfer1::Weights >	mGateBiasWeights
bool	mRNNKind
int	mNumLayers
int	mNumUnits

Detailed Description

encodes single input into output states with LSTM

Member Typedef Documentation

ptr

typedef std::shared_ptr<Decoder> nmtSample::Decoder::ptr

inherited

Constructor & Destructor Documentation

LSTMDecoder()

nmtSample::LSTMDecoder::LSTMDecoder

(

ComponentWeights::ptr

weights

)

Here is the call graph for this function:

~LSTMDecoder()

nmtSample::LSTMDecoder::~LSTMDecoder ( )

overridedefault

Member Function Documentation

addToModel()

void nmtSample::LSTMDecoder::addToModel ( nvinfer1::INetworkDefinition *  network, nvinfer1::ITensor *  inputData, nvinfer1::ITensor **  inputStates, nvinfer1::ITensor **  outputData, nvinfer1::ITensor **  outputStates  )

overridevirtual

add the memory, cell, and hidden states to the network

Implements nmtSample::Decoder.

Here is the call graph for this function:

getStateSizes()

std::vector< nvinfer1::Dims > nmtSample::LSTMDecoder::getStateSizes ( )

overridevirtual

get the sizes (vector of them) of the hidden state vectors

Implements nmtSample::Decoder.

getInfo()

std::string nmtSample::LSTMDecoder::getInfo ( )

overridevirtual

get the textual description of the component

Implements nmtSample::Component.

Member Data Documentation

mWeights

ComponentWeights::ptr nmtSample::LSTMDecoder::mWeights

protected

mGateKernelWeights

std::vector<nvinfer1::Weights> nmtSample::LSTMDecoder::mGateKernelWeights

protected

mGateBiasWeights

std::vector<nvinfer1::Weights> nmtSample::LSTMDecoder::mGateBiasWeights

protected

mRNNKind

bool nmtSample::LSTMDecoder::mRNNKind

protected

mNumLayers

int nmtSample::LSTMDecoder::mNumLayers

protected

mNumUnits

int nmtSample::LSTMDecoder::mNumUnits

protected

---

The documentation for this class was generated from the following files:

• lstmDecoder.h
• lstmDecoder.cpp