ProposalWorkSpace Struct Reference

Collaboration diagram for ProposalWorkSpace:

Public Member Functions
	ProposalWorkSpace (const int batchSize, const int inputCnt, const int sampleCount, const RefineNMSParameters &param, const nvinfer1::DataType type)
	ProposalWorkSpace ()=default

Public Attributes
nvinfer1::DimsHW	preRefineScoreDims
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	preRefineScoreOffset = 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

ProposalWorkSpace() [1/2]

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

ProposalWorkSpace() [2/2]

ProposalWorkSpace::ProposalWorkSpace ( )

default

Member Data Documentation

preRefineScoreDims

nvinfer1::DimsHW ProposalWorkSpace::preRefineScoreDims

preRefineSortedScoreDims

nvinfer1::DimsHW ProposalWorkSpace::preRefineSortedScoreDims

preRefineBboxDims

nvinfer1::DimsHW ProposalWorkSpace::preRefineBboxDims

argMaxScoreDims

nvinfer1::DimsHW ProposalWorkSpace::argMaxScoreDims

argMaxBboxDims

nvinfer1::DimsHW ProposalWorkSpace::argMaxBboxDims

argMaxLabelDims

nvinfer1::DimsHW ProposalWorkSpace::argMaxLabelDims

sortClassScoreDims

nvinfer1::DimsHW ProposalWorkSpace::sortClassScoreDims

sortClassLabelDims

nvinfer1::DimsHW ProposalWorkSpace::sortClassLabelDims

sortClassSampleIdxDims

nvinfer1::DimsHW ProposalWorkSpace::sortClassSampleIdxDims

sortClassValidCountDims

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

sortClassPosDims

nvinfer1::DimsHW ProposalWorkSpace::sortClassPosDims

sortNMSMarkDims

nvinfer1::DimsHW ProposalWorkSpace::sortNMSMarkDims

tempStorageOffset

size_t ProposalWorkSpace::tempStorageOffset = 0

preRefineScoreOffset

size_t ProposalWorkSpace::preRefineScoreOffset = 0

preRefineSortedScoreOffset

size_t ProposalWorkSpace::preRefineSortedScoreOffset = 0

preRefineBboxOffset

size_t ProposalWorkSpace::preRefineBboxOffset = 0

argMaxScoreOffset

size_t ProposalWorkSpace::argMaxScoreOffset = 0

argMaxBboxOffset

size_t ProposalWorkSpace::argMaxBboxOffset = 0

argMaxLabelOffset

size_t ProposalWorkSpace::argMaxLabelOffset = 0

sortClassScoreOffset

size_t ProposalWorkSpace::sortClassScoreOffset = 0

sortClassLabelOffset

size_t ProposalWorkSpace::sortClassLabelOffset = 0

sortClassSampleIdxOffset

size_t ProposalWorkSpace::sortClassSampleIdxOffset = 0

sortClassValidCountOffset

size_t ProposalWorkSpace::sortClassValidCountOffset = 0

sortClassPosOffset

size_t ProposalWorkSpace::sortClassPosOffset = 0

sortNMSMarkOffset

size_t ProposalWorkSpace::sortNMSMarkOffset = 0

totalSize

size_t ProposalWorkSpace::totalSize = 0

---

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

• maskRCNNKernels.h