nvdsinfer::TrtEngine Class Reference

Detailed Description

Helper class written on top of nvinfer1::ICudaEngine.

Definition at line 194 of file 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-> ()

Constructor & Destructor Documentation

TrtEngine() [1/2]

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

inline

Definition at line 197 of file nvdsinfer_model_builder.h.

TrtEngine() [2/2]

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()

nvdsinfer::TrtEngine::~TrtEngine

(

)

Member Function Documentation

engine()

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

inline

Definition at line 218 of file nvdsinfer_model_builder.h.

getDlaCore()

int nvdsinfer::TrtEngine::getDlaCore ( ) const

inline

Definition at line 208 of file nvdsinfer_model_builder.h.

getFullDimsLayersInfo()

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

getImplicitLayersInfo()

NvDsInferStatus nvdsinfer::TrtEngine::getImplicitLayersInfo

(

std::vector< NvDsInferBatchDimsLayerInfo > &

layersInfo

)

getLayerInfo()

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

hasDla()

bool nvdsinfer::TrtEngine::hasDla ( ) const

inline

Definition at line 207 of file nvdsinfer_model_builder.h.

operator->()

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

inline

Definition at line 224 of file nvdsinfer_model_builder.h.

printEngineInfo()

void nvdsinfer::TrtEngine::printEngineInfo

(

)

---

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

• nvdsinfer_model_builder.h