Searched refs:outputScore (Results 1 – 2 of 2) sorted by relevance
281 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare() local282 outputScore.type = heatmapShape.type; in prepare()283 outputScore.dimensions = {numBoxes, numKeypoints}; in prepare()284 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
3515 ANeuralNetworksOperandType outputScore = getOpType(inputOperandCode, 2, outputScoreDims); in detectionPostprocessingOpTest() local3527 {outputScore, boundingBoxes, classLabel, numValid}); in detectionPostprocessingOpTest()