Home
last modified time | relevance | path

Searched refs:outputScore (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DHeatmapMaxKeypoint.cpp244 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare() local
245 outputScore.type = heatmapShape.type; in prepare()
246 outputScore.dimensions = {numBoxes, numKeypoints}; in prepare()
247 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp3554 ANeuralNetworksOperandType outputScore = getOpType(inputOperandCode, 2, outputScoreDims); in detectionPostprocessingOpTest() local
3566 {outputScore, boundingBoxes, classLabel, numValid}); in detectionPostprocessingOpTest()