TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
nmtSample::DataWriter Class Referenceabstract

writer of sequences of data More...

Inheritance diagram for nmtSample::DataWriter:
Collaboration diagram for nmtSample::DataWriter:

Public Types

typedef std::shared_ptr< DataWriterptr
 

Public Member Functions

 DataWriter ()=default
 
virtual void write (const int *hOutputData, int actualOutputSequenceLength, int actualInputSequenceLength)=0
 write the generated sequence More...
 
virtual void initialize ()=0
 it is called right before inference starts More...
 
virtual void finalize ()=0
 it is called right after inference ends More...
 
 ~DataWriter () override=default
 
virtual std::string getInfo ()=0
 get the textual description of the component More...
 

Static Protected Member Functions

static std::string generateText (int sequenceLength, const int *currentOutputData, Vocabulary::ptr vocabulary)
 

Detailed Description

writer of sequences of data

Member Typedef Documentation

◆ ptr

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

Constructor & Destructor Documentation

◆ DataWriter()

nmtSample::DataWriter::DataWriter ( )
default

◆ ~DataWriter()

nmtSample::DataWriter::~DataWriter ( )
overridedefault

Member Function Documentation

◆ write()

virtual void nmtSample::DataWriter::write ( const int hOutputData,
int  actualOutputSequenceLength,
int  actualInputSequenceLength 
)
pure virtual

write the generated sequence

Implemented in nmtSample::BLEUScoreWriter, nmtSample::TextWriter, and nmtSample::BenchmarkWriter.

◆ initialize()

virtual void nmtSample::DataWriter::initialize ( )
pure virtual

it is called right before inference starts

Implemented in nmtSample::BLEUScoreWriter, nmtSample::TextWriter, and nmtSample::BenchmarkWriter.

◆ finalize()

virtual void nmtSample::DataWriter::finalize ( )
pure virtual

it is called right after inference ends

Implemented in nmtSample::BLEUScoreWriter, nmtSample::TextWriter, and nmtSample::BenchmarkWriter.

◆ generateText()

std::string nmtSample::DataWriter::generateText ( int  sequenceLength,
const int currentOutputData,
Vocabulary::ptr  vocabulary 
)
staticprotected
Here is the caller graph for this function:

◆ getInfo()


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