nvdsinfer::BuildParams Struct Reference

Detailed Description

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

Definition at line 134 of file 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 >

Public Member Functions
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

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

Definition at line 139 of file nvdsinfer_model_builder.h.

TensorIOFormat

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

Definition at line 137 of file nvdsinfer_model_builder.h.

Constructor & Destructor Documentation

~BuildParams()

virtual nvdsinfer::BuildParams::~BuildParams ( )

inlinevirtual

Definition at line 150 of file nvdsinfer_model_builder.h.

Member Function Documentation

configBuilder()

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

pure virtual

sanityCheck()

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

virtual

Field Documentation

dlaCore

int nvdsinfer::BuildParams::dlaCore = -1

Definition at line 144 of file nvdsinfer_model_builder.h.

inputFormats

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

Definition at line 145 of file nvdsinfer_model_builder.h.

int8CalibrationFilePath

std::string nvdsinfer::BuildParams::int8CalibrationFilePath

Definition at line 143 of file nvdsinfer_model_builder.h.

layerDevicePrecisions

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

Definition at line 147 of file nvdsinfer_model_builder.h.

networkMode

NvDsInferNetworkMode nvdsinfer::BuildParams::networkMode = NvDsInferNetworkMode_FP32

Definition at line 142 of file nvdsinfer_model_builder.h.

outputFormats

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

Definition at line 146 of file nvdsinfer_model_builder.h.

workspaceSize

size_t nvdsinfer::BuildParams::workspaceSize = kWorkSpaceSize

Definition at line 141 of file nvdsinfer_model_builder.h.

---

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

• nvdsinfer_model_builder.h