nmtSample::SoftmaxLikelihood Class Reference

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

Inheritance diagram for nmtSample::SoftmaxLikelihood:

Collaboration diagram for nmtSample::SoftmaxLikelihood:

Classes
class	SoftmaxLikelihoodCombinationOperator

Public Types
typedef std::shared_ptr< Likelihood >	ptr

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

Detailed Description

calculates softmax likelihood and TopK indices for the raw input logits

Member Typedef Documentation

ptr

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

inherited

Constructor & Destructor Documentation

SoftmaxLikelihood()

nmtSample::SoftmaxLikelihood::SoftmaxLikelihood ( )

default

~SoftmaxLikelihood()

nmtSample::SoftmaxLikelihood::~SoftmaxLikelihood ( )

overridedefault

Member Function Documentation

getLikelihoodCombinationOperator()

LikelihoodCombinationOperator::ptr nmtSample::SoftmaxLikelihood::getLikelihoodCombinationOperator ( ) const

overridevirtual

Implements nmtSample::Likelihood.

addToModel()

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

overridevirtual

add calculation of likelihood and TopK indices to the network

Implements nmtSample::Likelihood.

Here is the call graph for this function:

getInfo()

std::string nmtSample::SoftmaxLikelihood::getInfo ( )

overridevirtual

get the textual description of the component

Implements nmtSample::Component.

---

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

• softmaxLikelihood.h
• softmaxLikelihood.cpp