Lines Matching refs:roiOperandCode
3190 void roiAlignOpTest(int32_t inputOperandCode, int32_t roiOperandCode, int32_t scalarOperandCode) { in roiAlignOpTest() argument
3195 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiAlignOpTest()
3225 void roiPoolingOpTest(int32_t inputOperandCode, int32_t roiOperandCode, int32_t scalarOperandCode) { in roiPoolingOpTest() argument
3230 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiPoolingOpTest()
3259 void heatmapMaxKeypointOpTest(int32_t heatmapOperandCode, int32_t roiOperandCode) { in heatmapMaxKeypointOpTest() argument
3264 {getOpType(heatmapOperandCode, 4, heatmapDim), getOpType(roiOperandCode, 2, boxDim), in heatmapMaxKeypointOpTest()
3267 getOpType(roiOperandCode, 3, outKeypointDim)}); in heatmapMaxKeypointOpTest()
3620 void axisAlignedBBoxTransformOpTest(int32_t roiOperandCode, int32_t deltaOperandCode) { in axisAlignedBBoxTransformOpTest() argument
3625 {getOpType(roiOperandCode, 2, roiDim), getOpType(deltaOperandCode, 2, deltaDim), in axisAlignedBBoxTransformOpTest()
3627 getOpType(roiOperandCode, 2, imageDim)}, in axisAlignedBBoxTransformOpTest()
3628 {getOpType(roiOperandCode, 2, outDim)}); in axisAlignedBBoxTransformOpTest()
3849 void boxWithNmsLimitOpTest(int32_t scoreOperandCode, int32_t roiOperandCode, in boxWithNmsLimitOpTest() argument
3855 {getOpType(scoreOperandCode, 2, scoreDim), getOpType(roiOperandCode, 2, roiDim), in boxWithNmsLimitOpTest()
3860 {getOpType(scoreOperandCode, 1, outScoreDim), getOpType(roiOperandCode, 2, outRoiDim), in boxWithNmsLimitOpTest()
4223 int32_t anchorOperandCode, int32_t roiOperandCode, in generateProposalsOpTest() argument
4231 getOpType(anchorOperandCode, 2, anchorDim), getOpType(roiOperandCode, 2, imageInfoDim), in generateProposalsOpTest()
4236 {getOpType(scoreOperandCode, 1, outScoreDim), getOpType(roiOperandCode, 2, outRoiDim), in generateProposalsOpTest()