nmtSample::SLPEmbedder Class Reference

selects the embedding vector from the weight matrix using index provided in the input More...

Inheritance diagram for nmtSample::SLPEmbedder:

Collaboration diagram for nmtSample::SLPEmbedder:

Public Types
typedef std::shared_ptr< Embedder >	ptr

Public Member Functions
	SLPEmbedder (ComponentWeights::ptr weights)
void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *input, nvinfer1::ITensor **output) override
	add the embedding vector calculation to the network More...
int	getInputDimensionSize () override
	get the upper bound for the possible values of indices More...
std::string	getInfo () override
	get the textual description of the component More...
	~SLPEmbedder () override=default

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

Detailed Description

selects the embedding vector from the weight matrix using index provided in the input

Member Typedef Documentation

ptr

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

inherited

Constructor & Destructor Documentation

SLPEmbedder()

nmtSample::SLPEmbedder::SLPEmbedder

(

ComponentWeights::ptr

weights

)

Here is the call graph for this function:

~SLPEmbedder()

nmtSample::SLPEmbedder::~SLPEmbedder ( )

overridedefault

Member Function Documentation

addToModel()

void nmtSample::SLPEmbedder::addToModel ( nvinfer1::INetworkDefinition *  network, nvinfer1::ITensor *  input, nvinfer1::ITensor **  output  )

overridevirtual

add the embedding vector calculation to the network

Implements nmtSample::Embedder.

Here is the call graph for this function:

getInputDimensionSize()

int nmtSample::SLPEmbedder::getInputDimensionSize ( )

overridevirtual

get the upper bound for the possible values of indices

Implements nmtSample::Embedder.

getInfo()

std::string nmtSample::SLPEmbedder::getInfo ( )

overridevirtual

get the textual description of the component

Implements nmtSample::Component.

Member Data Documentation

mWeights

ComponentWeights::ptr nmtSample::SLPEmbedder::mWeights

protected

mKernelWeights

nvinfer1::Weights nmtSample::SLPEmbedder::mKernelWeights

protected

mNumInputs

int nmtSample::SLPEmbedder::mNumInputs

protected

mNumOutputs

int nmtSample::SLPEmbedder::mNumOutputs

protected

mResizedKernelWeights

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

protected

---

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

• slpEmbedder.h
• slpEmbedder.cpp