NVIDIA DeepStream SDK API Reference

8.0 Release
nvdsinfer::TrtEngine Class Reference

Detailed Description

Helper class written on top of nvinfer1::ICudaEngine.

Definition at line 194 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

Public Member Functions

 TrtEngine (std::unique_ptr< nvinfer1::ICudaEngine > &&engine, int dlaCore=-1)
 
 TrtEngine (std::unique_ptr< nvinfer1::ICudaEngine > &&engine, const std::shared_ptr< nvinfer1::IRuntime > &runtime, int dlaCore=-1, const std::shared_ptr< DlLibHandle > &dlHandle=nullptr, nvinfer1::IPluginFactory *pluginFactory=nullptr)
 
 ~TrtEngine ()
 
bool hasDla () const
 
int getDlaCore () const
 
NvDsInferStatus getImplicitLayersInfo (std::vector< NvDsInferBatchDimsLayerInfo > &layersInfo)
 
NvDsInferStatus getFullDimsLayersInfo (int profileIdx, std::vector< NvDsInferBatchDimsLayerInfo > &layersInfo)
 
NvDsInferStatus getLayerInfo (int idx, NvDsInferLayerInfo &layer)
 
void printEngineInfo ()
 
nvinfer1::ICudaEngine & engine ()
 
nvinfer1::ICudaEngine * operator-> ()
 
 TrtEngine (std::unique_ptr< nvinfer1::ICudaEngine > &&engine, int dlaCore=-1)
 
 TrtEngine (std::unique_ptr< nvinfer1::ICudaEngine > &&engine, const std::shared_ptr< nvinfer1::IRuntime > &runtime, int dlaCore=-1, const std::shared_ptr< DlLibHandle > &dlHandle=nullptr, nvinfer1::IPluginFactory *pluginFactory=nullptr)
 
 ~TrtEngine ()
 
bool hasDla () const
 
int getDlaCore () const
 
NvDsInferStatus getImplicitLayersInfo (std::vector< NvDsInferBatchDimsLayerInfo > &layersInfo)
 
NvDsInferStatus getFullDimsLayersInfo (int profileIdx, std::vector< NvDsInferBatchDimsLayerInfo > &layersInfo)
 
NvDsInferStatus getLayerInfo (int idx, NvDsInferLayerInfo &layer)
 
void printEngineInfo ()
 
nvinfer1::ICudaEngine & engine ()
 
nvinfer1::ICudaEngine * operator-> ()
 

Constructor & Destructor Documentation

◆ TrtEngine() [1/4]

nvdsinfer::TrtEngine::TrtEngine ( std::unique_ptr< nvinfer1::ICudaEngine > &&  engine,
int  dlaCore = -1 
)
inline

◆ TrtEngine() [2/4]

nvdsinfer::TrtEngine::TrtEngine ( std::unique_ptr< nvinfer1::ICudaEngine > &&  engine,
const std::shared_ptr< nvinfer1::IRuntime > &  runtime,
int  dlaCore = -1,
const std::shared_ptr< DlLibHandle > &  dlHandle = nullptr,
nvinfer1::IPluginFactory *  pluginFactory = nullptr 
)

◆ ~TrtEngine() [1/2]

nvdsinfer::TrtEngine::~TrtEngine ( )

◆ TrtEngine() [3/4]

nvdsinfer::TrtEngine::TrtEngine ( std::unique_ptr< nvinfer1::ICudaEngine > &&  engine,
int  dlaCore = -1 
)
inline

◆ TrtEngine() [4/4]

nvdsinfer::TrtEngine::TrtEngine ( std::unique_ptr< nvinfer1::ICudaEngine > &&  engine,
const std::shared_ptr< nvinfer1::IRuntime > &  runtime,
int  dlaCore = -1,
const std::shared_ptr< DlLibHandle > &  dlHandle = nullptr,
nvinfer1::IPluginFactory *  pluginFactory = nullptr 
)

◆ ~TrtEngine() [2/2]

nvdsinfer::TrtEngine::~TrtEngine ( )

Member Function Documentation

◆ engine() [1/2]

nvinfer1::ICudaEngine& nvdsinfer::TrtEngine::engine ( )
inline

◆ engine() [2/2]

nvinfer1::ICudaEngine& nvdsinfer::TrtEngine::engine ( )
inline

◆ getDlaCore() [1/2]

int nvdsinfer::TrtEngine::getDlaCore ( ) const
inline

◆ getDlaCore() [2/2]

int nvdsinfer::TrtEngine::getDlaCore ( ) const
inline

◆ getFullDimsLayersInfo() [1/2]

NvDsInferStatus nvdsinfer::TrtEngine::getFullDimsLayersInfo ( int  profileIdx,
std::vector< NvDsInferBatchDimsLayerInfo > &  layersInfo 
)

◆ getFullDimsLayersInfo() [2/2]

NvDsInferStatus nvdsinfer::TrtEngine::getFullDimsLayersInfo ( int  profileIdx,
std::vector< NvDsInferBatchDimsLayerInfo > &  layersInfo 
)

◆ getImplicitLayersInfo() [1/2]

NvDsInferStatus nvdsinfer::TrtEngine::getImplicitLayersInfo ( std::vector< NvDsInferBatchDimsLayerInfo > &  layersInfo)

◆ getImplicitLayersInfo() [2/2]

NvDsInferStatus nvdsinfer::TrtEngine::getImplicitLayersInfo ( std::vector< NvDsInferBatchDimsLayerInfo > &  layersInfo)

◆ getLayerInfo() [1/2]

NvDsInferStatus nvdsinfer::TrtEngine::getLayerInfo ( int  idx,
NvDsInferLayerInfo layer 
)

◆ getLayerInfo() [2/2]

NvDsInferStatus nvdsinfer::TrtEngine::getLayerInfo ( int  idx,
NvDsInferLayerInfo layer 
)

◆ hasDla() [1/2]

bool nvdsinfer::TrtEngine::hasDla ( ) const
inline

◆ hasDla() [2/2]

bool nvdsinfer::TrtEngine::hasDla ( ) const
inline

◆ operator->() [1/2]

nvinfer1::ICudaEngine* nvdsinfer::TrtEngine::operator-> ( )
inline

◆ operator->() [2/2]

nvinfer1::ICudaEngine* nvdsinfer::TrtEngine::operator-> ( )
inline

◆ printEngineInfo() [1/2]

void nvdsinfer::TrtEngine::printEngineInfo ( )

◆ printEngineInfo() [2/2]

void nvdsinfer::TrtEngine::printEngineInfo ( )

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