TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
MultilevelProposeROIWorkSpace Struct Reference
Collaboration diagram for MultilevelProposeROIWorkSpace:

Public Member Functions

 MultilevelProposeROIWorkSpace (const int batchSize, const int inputCnt, const int sampleCount, const RefineNMSParameters &param, const nvinfer1::DataType type)
 
 MultilevelProposeROIWorkSpace ()=default
 

Public Attributes

nvinfer1::DimsHW preRefineSortedScoreDims
 
nvinfer1::DimsHW preRefineBboxDims
 
nvinfer1::DimsHW argMaxScoreDims
 
nvinfer1::DimsHW argMaxBboxDims
 
nvinfer1::DimsHW argMaxLabelDims
 
nvinfer1::DimsHW sortClassScoreDims
 
nvinfer1::DimsHW sortClassLabelDims
 
nvinfer1::DimsHW sortClassSampleIdxDims
 
nvinfer1::Dims sortClassValidCountDims = {1, {1, 0}, {nvinfer1::DimensionType::kINDEX}}
 
nvinfer1::DimsHW sortClassPosDims
 
nvinfer1::DimsHW sortNMSMarkDims
 
size_t tempStorageOffset = 0
 
size_t preRefineSortedScoreOffset = 0
 
size_t preRefineBboxOffset = 0
 
size_t argMaxScoreOffset = 0
 
size_t argMaxBboxOffset = 0
 
size_t argMaxLabelOffset = 0
 
size_t sortClassScoreOffset = 0
 
size_t sortClassLabelOffset = 0
 
size_t sortClassSampleIdxOffset = 0
 
size_t sortClassValidCountOffset = 0
 
size_t sortClassPosOffset = 0
 
size_t sortNMSMarkOffset = 0
 
size_t totalSize = 0
 

Constructor & Destructor Documentation

◆ MultilevelProposeROIWorkSpace() [1/2]

MultilevelProposeROIWorkSpace::MultilevelProposeROIWorkSpace ( const int  batchSize,
const int  inputCnt,
const int  sampleCount,
const RefineNMSParameters param,
const nvinfer1::DataType  type 
)

◆ MultilevelProposeROIWorkSpace() [2/2]

MultilevelProposeROIWorkSpace::MultilevelProposeROIWorkSpace ( )
default

Member Data Documentation

◆ preRefineSortedScoreDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::preRefineSortedScoreDims

◆ preRefineBboxDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::preRefineBboxDims

◆ argMaxScoreDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::argMaxScoreDims

◆ argMaxBboxDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::argMaxBboxDims

◆ argMaxLabelDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::argMaxLabelDims

◆ sortClassScoreDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::sortClassScoreDims

◆ sortClassLabelDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::sortClassLabelDims

◆ sortClassSampleIdxDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::sortClassSampleIdxDims

◆ sortClassValidCountDims

nvinfer1::Dims MultilevelProposeROIWorkSpace::sortClassValidCountDims = {1, {1, 0}, {nvinfer1::DimensionType::kINDEX}}

◆ sortClassPosDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::sortClassPosDims

◆ sortNMSMarkDims

nvinfer1::DimsHW MultilevelProposeROIWorkSpace::sortNMSMarkDims

◆ tempStorageOffset

size_t MultilevelProposeROIWorkSpace::tempStorageOffset = 0

◆ preRefineSortedScoreOffset

size_t MultilevelProposeROIWorkSpace::preRefineSortedScoreOffset = 0

◆ preRefineBboxOffset

size_t MultilevelProposeROIWorkSpace::preRefineBboxOffset = 0

◆ argMaxScoreOffset

size_t MultilevelProposeROIWorkSpace::argMaxScoreOffset = 0

◆ argMaxBboxOffset

size_t MultilevelProposeROIWorkSpace::argMaxBboxOffset = 0

◆ argMaxLabelOffset

size_t MultilevelProposeROIWorkSpace::argMaxLabelOffset = 0

◆ sortClassScoreOffset

size_t MultilevelProposeROIWorkSpace::sortClassScoreOffset = 0

◆ sortClassLabelOffset

size_t MultilevelProposeROIWorkSpace::sortClassLabelOffset = 0

◆ sortClassSampleIdxOffset

size_t MultilevelProposeROIWorkSpace::sortClassSampleIdxOffset = 0

◆ sortClassValidCountOffset

size_t MultilevelProposeROIWorkSpace::sortClassValidCountOffset = 0

◆ sortClassPosOffset

size_t MultilevelProposeROIWorkSpace::sortClassPosOffset = 0

◆ sortNMSMarkOffset

size_t MultilevelProposeROIWorkSpace::sortNMSMarkOffset = 0

◆ totalSize

size_t MultilevelProposeROIWorkSpace::totalSize = 0

The documentation for this struct was generated from the following file: