nmtSample::SLPProjection Class Reference

Linear logits calculation. More...

Inheritance diagram for nmtSample::SLPProjection:

Collaboration diagram for nmtSample::SLPProjection:

Public Types
typedef std::shared_ptr< Projection >	ptr

Public Member Functions
	SLPProjection (ComponentWeights::ptr weights)
void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *input, nvinfer1::ITensor **outputLogits) override
	add raw logits to the network More...
int	getOutputSize () override
	get the size of raw logits vector More...
std::string	getInfo () override
	get the textual description of the component More...
	~SLPProjection () override=default

Protected Attributes
ComponentWeights::ptr	mWeights
nvinfer1::Weights	mKernelWeights
int	mInputChannelCount
int	mOutputChannelCount
std::vector< float >	mResizedKernelWeights

Detailed Description

Linear logits calculation.

Calculates logits vector by multiplying input vector with weight matrix

Member Typedef Documentation

ptr

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

inherited

Constructor & Destructor Documentation

SLPProjection()

nmtSample::SLPProjection::SLPProjection

(

ComponentWeights::ptr

weights

)

Here is the call graph for this function:

~SLPProjection()

nmtSample::SLPProjection::~SLPProjection ( )

overridedefault

Member Function Documentation

addToModel()

void nmtSample::SLPProjection::addToModel ( nvinfer1::INetworkDefinition *  network, nvinfer1::ITensor *  input, nvinfer1::ITensor **  outputLogits  )

overridevirtual

add raw logits to the network

Implements nmtSample::Projection.

Here is the call graph for this function:

getOutputSize()

int nmtSample::SLPProjection::getOutputSize ( )

overridevirtual

get the size of raw logits vector

Implements nmtSample::Projection.

getInfo()

std::string nmtSample::SLPProjection::getInfo ( )

overridevirtual

get the textual description of the component

Implements nmtSample::Component.

Member Data Documentation

mWeights

ComponentWeights::ptr nmtSample::SLPProjection::mWeights

protected

mKernelWeights

nvinfer1::Weights nmtSample::SLPProjection::mKernelWeights

protected

mInputChannelCount

int nmtSample::SLPProjection::mInputChannelCount

protected

mOutputChannelCount

int nmtSample::SLPProjection::mOutputChannelCount

protected

mResizedKernelWeights

std::vector<float> nmtSample::SLPProjection::mResizedKernelWeights

protected

---

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

• slpProjection.h
• slpProjection.cpp