Holds build parameters required for full dimensions network.
Definition at line 175 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 139 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 137 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 144 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 145 of file nvdsinfer_model_builder.h.
NvDsInferTensorOrder nvdsinfer::ExplicitBuildParams::inputOrder = NvDsInferTensorOrder_kNCHW |
Definition at line 184 of file nvdsinfer_model_builder.h.
std::vector<ProfileDims> nvdsinfer::ExplicitBuildParams::inputProfileDims |
Definition at line 180 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 143 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 147 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::maxBatchSize = 1 |
Definition at line 183 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::minBatchSize = 1 |
Definition at line 181 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 142 of file nvdsinfer_model_builder.h.
int nvdsinfer::ExplicitBuildParams::optBatchSize = 1 |
Definition at line 182 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 146 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 141 of file nvdsinfer_model_builder.h.