Holds build parameters required for implicit batch dimension network.
Definition at line 158 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 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.
std::vector<nvinfer1::Dims> nvdsinfer::ImplicitBuildParams::inputDims |
Definition at line 161 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 145 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::ImplicitBuildParams::maxBatchSize = 0 |
Definition at line 160 of file nvdsinfer_model_builder.h.
|
inherited |
Definition at line 142 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.