nmtSample::Context Class Reference

calculates context vector from raw alignment scores and memory states More...

Inheritance diagram for nmtSample::Context:

Collaboration diagram for nmtSample::Context:

Public Types
typedef std::shared_ptr< Context >	ptr

Public Member Functions
	Context ()=default
void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *actualInputSequenceLengths, nvinfer1::ITensor *memoryStates, nvinfer1::ITensor *alignmentScores, nvinfer1::ITensor **contextOutput)
	add the context vector calculation to the network More...
std::string	getInfo () override
	get the textual description of the component More...
	~Context () override=default

Detailed Description

calculates context vector from raw alignment scores and memory states

Member Typedef Documentation

ptr

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

Constructor & Destructor Documentation

Context()

nmtSample::Context::Context ( )

default

~Context()

nmtSample::Context::~Context ( )

overridedefault

Member Function Documentation

addToModel()

void nmtSample::Context::addToModel	(	nvinfer1::INetworkDefinition *	network,
		nvinfer1::ITensor *	actualInputSequenceLengths,
		nvinfer1::ITensor *	memoryStates,
		nvinfer1::ITensor *	alignmentScores,
		nvinfer1::ITensor **	contextOutput
	)

add the context vector calculation to the network

Here is the call graph for this function:

getInfo()

std::string nmtSample::Context::getInfo ( )

overridevirtual

get the textual description of the component

Implements nmtSample::Component.

---

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

• contextNMT.h
• contextNMT.cpp