Searched refs:roiDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3190 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiAlignOpTest() local
3194 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiAlignOpTest()
3225 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiPoolingOpTest() local
3229 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiPoolingOpTest()
3620 uint32_t roiDim[] = {5, 4}, deltaDim[] = {5, 8}, bsDim[] = {5}, imageDim[] = {5, 2}; in axisAlignedBBoxTransformOpTest() local
3624 {getOpType(roiOperandCode, 2, roiDim), getOpType(deltaOperandCode, 2, deltaDim), in axisAlignedBBoxTransformOpTest()
3850 uint32_t scoreDim[] = {19, 3}, roiDim[] = {19, 12}, splitDim[] = {2}; in boxWithNmsLimitOpTest() local
3854 {getOpType(scoreOperandCode, 2, scoreDim), getOpType(roiOperandCode, 2, roiDim), in boxWithNmsLimitOpTest()
|