Searched refs:roiDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3191 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiAlignOpTest() local
3195 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiAlignOpTest()
3226 uint32_t inDim[] = {1, 4, 4, 1}, roiDim[] = {4, 4}, batchSplitDim[] = {1}; in roiPoolingOpTest() local
3230 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiPoolingOpTest()
3621 uint32_t roiDim[] = {5, 4}, deltaDim[] = {5, 8}, bsDim[] = {5}, imageDim[] = {5, 2}; in axisAlignedBBoxTransformOpTest() local
3625 {getOpType(roiOperandCode, 2, roiDim), getOpType(deltaOperandCode, 2, deltaDim), in axisAlignedBBoxTransformOpTest()
3851 uint32_t scoreDim[] = {19, 3}, roiDim[] = {19, 12}, splitDim[] = {2}; in boxWithNmsLimitOpTest() local
3855 {getOpType(scoreOperandCode, 2, scoreDim), getOpType(roiOperandCode, 2, roiDim), in boxWithNmsLimitOpTest()
|