Holds build parameters required for full dimensions network.
Definition at line 236 of file nvdsinfer_model_builder.h.
Public Types | |
using | TensorIOFormat = std::tuple< nvinfer1::DataType, nvinfer1::TensorFormats > |
using | LayerDevicePrecision = std::tuple< nvinfer1::DataType, nvinfer1::DeviceType > |
Data Fields | |
std::vector< ProfileDims > | inputProfileDims |
int | minBatchSize = 1 |
int | optBatchSize = 1 |
int | maxBatchSize = 1 |
NvDsInferTensorOrder | inputOrder = NvDsInferTensorOrder_kNCHW |
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 |
|
inherited |
Definition at line 200 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 198 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 205 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 206 of file nvdsinfer_model_builder.h.
NvDsInferTensorOrder nvdsinfer::ExplicitBuildParams::inputOrder = NvDsInferTensorOrder_kNCHW |
Definition at line 245 of file nvdsinfer_model_builder.h.
std::vector<ProfileDims> nvdsinfer::ExplicitBuildParams::inputProfileDims |
Definition at line 241 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 204 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 208 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::maxBatchSize = 1 |
Definition at line 244 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::minBatchSize = 1 |
Definition at line 242 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 203 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::optBatchSize = 1 |
Definition at line 243 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 207 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 202 of file nvdsinfer_model_builder.h.