Holds build parameters required for implicit batch dimension network.
Definition at line 219 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 | |
int | maxBatchSize = 0 |
std::vector< nvinfer1::Dims > | inputDims |
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.
std::vector<nvinfer1::Dims> nvdsinfer::ImplicitBuildParams::inputDims |
Definition at line 222 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 206 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::ImplicitBuildParams::maxBatchSize = 0 |
Definition at line 221 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 203 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.