Searched refs:scores_float32 (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 626 std::vector<float> scores_float32(getNumberOfElements(scoresShape)); in boxWithNmsLimitFloat32() local
627 for (uint32_t i = 0; i < scores_float32.size(); i++) { in boxWithNmsLimitFloat32()
628 scores_float32[i] = scoresData[i]; in boxWithNmsLimitFloat32()
632 scores_float32.data(), scoresShape, roiData, roiShape, batchesData, batchesShape, in boxWithNmsLimitFloat32()
636 scores_float32, context); in boxWithNmsLimitFloat32()
649 std::vector<float> scores_float32(getNumberOfElements(scoresShape)); in boxWithNmsLimitFloat16() local
650 convertFloat16ToFloat32(scoresData, &scores_float32); in boxWithNmsLimitFloat16()
655 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitFloat16()
659 scores_float32, context); in boxWithNmsLimitFloat16()
671 std::vector<float> scores_float32(getNumberOfElements(scoresShape)); in boxWithNmsLimitQuant() local
[all …]
|