Helper class written on top of nvinfer1::ICudaEngine.
Definition at line 194 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.
|
| | 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-> () |
| |
◆ 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 |
( |
| ) |
|
◆ 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]
◆ getLayerInfo() [2/2]
◆ 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: