TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
sample::Profiler Class Reference

Collect per-layer profile information, assuming times are reported in the same order. More...

Inheritance diagram for sample::Profiler:
Collaboration diagram for sample::Profiler:

Public Member Functions

void reportLayerTime (const char *layerName, float timeMs) override
 Layer time reporting callback. More...
 
void print (std::ostream &os) const
 
void exportJSONProfile (const std::string &fileName) const
 Export a profile to JSON file. More...
 

Private Member Functions

float getTotalTime () const
 

Private Attributes

std::vector< LayerProfilemLayers
 
std::vector< LayerProfile >::iterator mIterator {mLayers.begin()}
 
int mUpdatesCount {0}
 

Detailed Description

Collect per-layer profile information, assuming times are reported in the same order.

Member Function Documentation

◆ reportLayerTime()

void sample::Profiler::reportLayerTime ( const char *  layerName,
float  ms 
)
overridevirtual

Layer time reporting callback.

Parameters
layerNameThe name of the layer, set when constructing the network definition.
msThe time in milliseconds to execute the layer.

Implements nvinfer1::IProfiler.

◆ print()

void sample::Profiler::print ( std::ostream &  os) const

◆ exportJSONProfile()

void sample::Profiler::exportJSONProfile ( const std::string &  fileName) const

Export a profile to JSON file.

◆ getTotalTime()

float sample::Profiler::getTotalTime ( ) const
inlineprivate

Member Data Documentation

◆ mLayers

std::vector<LayerProfile> sample::Profiler::mLayers
private

◆ mIterator

std::vector<LayerProfile>::iterator sample::Profiler::mIterator {mLayers.begin()}
private

◆ mUpdatesCount

int sample::Profiler::mUpdatesCount {0}
private

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