nmtSample::Likelihood Class Reference

calculates likelihood and TopK indices for the raw input logits More...

Inheritance diagram for nmtSample::Likelihood:

Collaboration diagram for nmtSample::Likelihood:

Public Types
typedef std::shared_ptr< Likelihood >	ptr

Public Member Functions
	Likelihood ()=default
virtual LikelihoodCombinationOperator::ptr	getLikelihoodCombinationOperator () const =0
virtual void	addToModel (nvinfer1::INetworkDefinition *network, int beamWidth, nvinfer1::ITensor *inputLogits, nvinfer1::ITensor *inputLikelihoods, nvinfer1::ITensor **newCombinedLikelihoods, nvinfer1::ITensor **newRayOptionIndices, nvinfer1::ITensor **newVocabularyIndices)=0
	add calculation of likelihood and TopK indices to the network More...
	~Likelihood () override=default
virtual std::string	getInfo ()=0
	get the textual description of the component More...

Detailed Description

calculates likelihood and TopK indices for the raw input logits

Member Typedef Documentation

ptr

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

Constructor & Destructor Documentation

Likelihood()

nmtSample::Likelihood::Likelihood ( )

default

~Likelihood()

nmtSample::Likelihood::~Likelihood ( )

overridedefault

Member Function Documentation

getLikelihoodCombinationOperator()

virtual LikelihoodCombinationOperator::ptr nmtSample::Likelihood::getLikelihoodCombinationOperator ( ) const

pure virtual

Implemented in nmtSample::SoftmaxLikelihood.

addToModel()

virtual void nmtSample::Likelihood::addToModel ( nvinfer1::INetworkDefinition *  network, int  beamWidth, nvinfer1::ITensor *  inputLogits, nvinfer1::ITensor *  inputLikelihoods, nvinfer1::ITensor **  newCombinedLikelihoods, nvinfer1::ITensor **  newRayOptionIndices, nvinfer1::ITensor **  newVocabularyIndices  )

pure virtual

add calculation of likelihood and TopK indices to the network

Implemented in nmtSample::SoftmaxLikelihood.

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:

• likelihood.h