nmtSample::Decoder Class Reference

encodes single input into output states More...

Inheritance diagram for nmtSample::Decoder:

Collaboration diagram for nmtSample::Decoder:

Public Types
typedef std::shared_ptr< Decoder >	ptr

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

Detailed Description

encodes single input into output states

Member Typedef Documentation

ptr

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

Constructor & Destructor Documentation

Decoder()

nmtSample::Decoder::Decoder ( )

default

~Decoder()

nmtSample::Decoder::~Decoder ( )

overridedefault

Member Function Documentation

addToModel()

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

pure virtual

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

Implemented in nmtSample::LSTMDecoder.

getStateSizes()

virtual std::vector<nvinfer1::Dims> nmtSample::Decoder::getStateSizes ( )

pure virtual

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

Implemented in nmtSample::LSTMDecoder.

getInfo()

virtual std::string nmtSample::Component::getInfo ( )

pure virtualinherited

get the textual description of the component

Implemented in nmtSample::SoftmaxLikelihood, nmtSample::BeamSearchPolicy, nmtSample::MultiplicativeAlignment, nmtSample::SLPEmbedder, nmtSample::SLPProjection, nmtSample::BLEUScoreWriter, nmtSample::Context, nmtSample::LSTMEncoder, nmtSample::TextWriter, nmtSample::SLPAttention, nmtSample::BenchmarkWriter, nmtSample::TextReader, nmtSample::LSTMDecoder, and nmtSample::LimitedSamplesDataReader.

---

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

• decoder.h