Searched refs:roiOperandCode (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 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()
[all …]
|