Searched defs:scoresOutShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 579 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput() local
625 Shape scoresOutShape, float* roiOutData, Shape roiOutShape, in boxWithNmsLimitFloat32()
648 const Shape& scoresOutShape, _Float16* roiOutData, in boxWithNmsLimitFloat16()
670 uint8_t* scoresOutData, const Shape& scoresOutShape, uint16_t* roiOutData, in boxWithNmsLimitQuant()
692 int8_t* scoresOutData, const Shape& scoresOutShape, uint16_t* roiOutData, in boxWithNmsLimitQuant()
1097 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat32() local
1149 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat16() local
1201 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsQuant() local
|