Holds build parameters common to implicit batch dimension/full dimension networks.
Definition at line 195 of file nvdsinfer_model_builder.h.
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::unordered_map< std::string, TensorIOFormat > | inputFormats |
std::unordered_map< std::string, TensorIOFormat > | outputFormats |
std::unordered_map< std::string, LayerDevicePrecision > | layerDevicePrecisions |
using nvdsinfer::BuildParams::LayerDevicePrecision = std::tuple<nvinfer1::DataType, nvinfer1::DeviceType> |
Definition at line 200 of file nvdsinfer_model_builder.h.
using nvdsinfer::BuildParams::TensorIOFormat = std::tuple<nvinfer1::DataType, nvinfer1::TensorFormats> |
Definition at line 198 of file nvdsinfer_model_builder.h.
|
inlinevirtual |
Definition at line 211 of file nvdsinfer_model_builder.h.
|
pure virtual |
|
virtual |
int nvdsinfer::BuildParams::dlaCore = -1 |
Definition at line 205 of file nvdsinfer_model_builder.h.
std::unordered_map<std::string, TensorIOFormat> nvdsinfer::BuildParams::inputFormats |
Definition at line 206 of file nvdsinfer_model_builder.h.
std::string nvdsinfer::BuildParams::int8CalibrationFilePath |
Definition at line 204 of file nvdsinfer_model_builder.h.
std::unordered_map<std::string, LayerDevicePrecision> nvdsinfer::BuildParams::layerDevicePrecisions |
Definition at line 208 of file nvdsinfer_model_builder.h.
NvDsInferNetworkMode nvdsinfer::BuildParams::networkMode = NvDsInferNetworkMode_FP32 |
Definition at line 203 of file nvdsinfer_model_builder.h.
std::unordered_map<std::string, TensorIOFormat> nvdsinfer::BuildParams::outputFormats |
Definition at line 207 of file nvdsinfer_model_builder.h.
size_t nvdsinfer::BuildParams::workspaceSize = kWorkSpaceSize |
Definition at line 202 of file nvdsinfer_model_builder.h.