nvdsinfer::BuildParams Struct Reference

Detailed Description

Holds build parameters common to implicit batch dimension/full dimension networks.

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

Inheritance diagram for nvdsinfer::BuildParams:

Public Types
using	TensorIOFormat = std::tuple< nvinfer1::DataType, nvinfer1::TensorFormats >
using	LayerDevicePrecision = std::tuple< nvinfer1::DataType, nvinfer1::DeviceType >
using	TensorIOFormat = std::tuple< nvinfer1::DataType, nvinfer1::TensorFormats >
using	LayerDevicePrecision = std::tuple< nvinfer1::DataType, nvinfer1::DeviceType >

Public Member Functions
virtual	~BuildParams ()
virtual NvDsInferStatus	configBuilder (TrtModelBuilder &builder)=0
virtual bool	sanityCheck () const
virtual	~BuildParams ()
virtual NvDsInferStatus	configBuilder (TrtModelBuilder &builder)=0
virtual bool	sanityCheck () const

Data Fields
size_t	workspaceSize = kWorkSpaceSize
NvDsInferNetworkMode	networkMode = NvDsInferNetworkMode_FP32
std::string	int8CalibrationFilePath
int	dlaCore = -1
std::vector< TensorIOFormat >	inputFormats
std::unordered_map< std::string, TensorIOFormat >	outputFormats
std::unordered_map< std::string, LayerDevicePrecision >	layerDevicePrecisions

Member Typedef Documentation

LayerDevicePrecision [1/2]

using nvdsinfer::BuildParams::LayerDevicePrecision = std::tuple<nvinfer1::DataType, nvinfer1::DeviceType>

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

LayerDevicePrecision [2/2]

using nvdsinfer::BuildParams::LayerDevicePrecision = std::tuple<nvinfer1::DataType, nvinfer1::DeviceType>

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

TensorIOFormat [1/2]

using nvdsinfer::BuildParams::TensorIOFormat = std::tuple<nvinfer1::DataType, nvinfer1::TensorFormats>

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

TensorIOFormat [2/2]

using nvdsinfer::BuildParams::TensorIOFormat = std::tuple<nvinfer1::DataType, nvinfer1::TensorFormats>

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

Constructor & Destructor Documentation

~BuildParams() [1/2]

virtual nvdsinfer::BuildParams::~BuildParams ( )

inlinevirtual

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

~BuildParams() [2/2]

virtual nvdsinfer::BuildParams::~BuildParams ( )

inlinevirtual

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

Member Function Documentation

configBuilder() [1/2]

virtual NvDsInferStatus nvdsinfer::BuildParams::configBuilder ( TrtModelBuilder &  builder )

pure virtual

configBuilder() [2/2]

virtual NvDsInferStatus nvdsinfer::BuildParams::configBuilder ( TrtModelBuilder &  builder )

pure virtual

sanityCheck() [1/2]

virtual bool nvdsinfer::BuildParams::sanityCheck ( ) const

virtual

sanityCheck() [2/2]

virtual bool nvdsinfer::BuildParams::sanityCheck ( ) const

virtual

Field Documentation

dlaCore

int nvdsinfer::BuildParams::dlaCore = -1

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

inputFormats

std::vector< TensorIOFormat > nvdsinfer::BuildParams::inputFormats

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

int8CalibrationFilePath

std::string nvdsinfer::BuildParams::int8CalibrationFilePath

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

layerDevicePrecisions

std::unordered_map< std::string, LayerDevicePrecision > nvdsinfer::BuildParams::layerDevicePrecisions

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

networkMode

NvDsInferNetworkMode nvdsinfer::BuildParams::networkMode = NvDsInferNetworkMode_FP32

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

outputFormats

std::unordered_map< std::string, TensorIOFormat > nvdsinfer::BuildParams::outputFormats

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

workspaceSize

size_t nvdsinfer::BuildParams::workspaceSize = kWorkSpaceSize

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

---

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

• sources/libs/nvdsinfer/nvdsinfer_model_builder.h