ConcatTopKWorkSpace Struct Reference

Collaboration diagram for ConcatTopKWorkSpace:

Public Member Functions
	ConcatTopKWorkSpace (const int batchSize, const int concatCnt, const int topK, const nvinfer1::DataType inType)
	ConcatTopKWorkSpace ()=default

Public Attributes
nvinfer1::DimsHW	concatedScoreDims
nvinfer1::DimsHW	concatedBBoxDims
nvinfer1::DimsHW	sortedScoreDims
nvinfer1::DimsHW	sortedBBoxDims
size_t	tempStorageOffset = 0
size_t	concatedScoreOffset = 0
size_t	concatedBBoxOffset = 0
size_t	sortedScoreOffset = 0
size_t	sortedBBoxOffset = 0
size_t	totalSize = 0

Constructor & Destructor Documentation

ConcatTopKWorkSpace() [1/2]

ConcatTopKWorkSpace::ConcatTopKWorkSpace	(	const int	batchSize,
		const int	concatCnt,
		const int	topK,
		const nvinfer1::DataType	inType
	)

ConcatTopKWorkSpace() [2/2]

ConcatTopKWorkSpace::ConcatTopKWorkSpace ( )

default

Member Data Documentation

concatedScoreDims

nvinfer1::DimsHW ConcatTopKWorkSpace::concatedScoreDims

concatedBBoxDims

nvinfer1::DimsHW ConcatTopKWorkSpace::concatedBBoxDims

sortedScoreDims

nvinfer1::DimsHW ConcatTopKWorkSpace::sortedScoreDims

sortedBBoxDims

nvinfer1::DimsHW ConcatTopKWorkSpace::sortedBBoxDims

tempStorageOffset

size_t ConcatTopKWorkSpace::tempStorageOffset = 0

concatedScoreOffset

size_t ConcatTopKWorkSpace::concatedScoreOffset = 0

concatedBBoxOffset

size_t ConcatTopKWorkSpace::concatedBBoxOffset = 0

sortedScoreOffset

size_t ConcatTopKWorkSpace::sortedScoreOffset = 0

sortedBBoxOffset

size_t ConcatTopKWorkSpace::sortedBBoxOffset = 0

totalSize

size_t ConcatTopKWorkSpace::totalSize = 0

---

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

• maskRCNNKernels.h