Searched refs:outScoreDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3260 uint32_t heatmapDim[] = {6, 4, 4, 1}, boxDim[] = {6, 4}, outScoreDim[] = {6, 1}, in heatmapMaxKeypointOpTest() local
3266 {getOpType(heatmapOperandCode, 2, outScoreDim), in heatmapMaxKeypointOpTest()
3852 uint32_t outScoreDim[] = {12}, outRoiDim[] = {12, 4}, outClassDim[] = {12}, outSplitDim[] = {2}; in boxWithNmsLimitOpTest() local
3860 {getOpType(scoreOperandCode, 1, outScoreDim), getOpType(roiOperandCode, 2, outRoiDim), in boxWithNmsLimitOpTest()
4227 uint32_t outScoreDim[] = {4}, outRoiDim[] = {4, 4}, outSplitDim[] = {1}; in generateProposalsOpTest() local
4236 {getOpType(scoreOperandCode, 1, outScoreDim), getOpType(roiOperandCode, 2, outRoiDim), in generateProposalsOpTest()
|