Lines Matching refs:Shape
60 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape, in bboxTransformFloat32()
61 const float* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat32()
62 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat32()
63 const float* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat32()
64 float* outputData, const Shape& outputShape) { in bboxTransformFloat32()
110 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16()
111 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat16()
112 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16()
113 const _Float16* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat16()
114 _Float16* outputData, const Shape& outputShape) { in bboxTransformFloat16()
130 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant()
131 const uint8_t* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformQuant()
132 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant()
133 const uint16_t* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformQuant()
134 uint16_t* outputData, const Shape& outputShape) { in bboxTransformQuant()
152 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant()
153 const int8_t* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformQuant()
154 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant()
155 const uint16_t* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformQuant()
156 uint16_t* outputData, const Shape& outputShape) { in bboxTransformQuant()
231 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
232 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
233 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
234 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
235 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
488 bool boxWithNmsLimitFloat32Compute(float* scoresData, const Shape& scoresShape, in boxWithNmsLimitFloat32Compute()
489 const float* roiData, const Shape& roiShape, in boxWithNmsLimitFloat32Compute()
490 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat32Compute()
553 T castTo(float val, const Shape&) { in castTo() argument
557 uint8_t castTo(float val, const Shape& shape) { in castTo()
562 int8_t castTo(float val, const Shape& shape) { in castTo()
573 Shape scoresShape = context->getInputShape(kScoreTensor); in boxWithNmsLimitWriteOutput()
579 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
583 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
587 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
591 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
620 bool boxWithNmsLimitFloat32(const float* scoresData, const Shape& scoresShape, const float* roiData, in boxWithNmsLimitFloat32()
621 const Shape& roiShape, const int32_t* batchesData, in boxWithNmsLimitFloat32()
622 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32()
625 Shape scoresOutShape, float* roiOutData, Shape roiOutShape, in boxWithNmsLimitFloat32()
626 int32_t* classesOutData, Shape classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat32()
627 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitFloat32()
642 bool boxWithNmsLimitFloat16(const _Float16* scoresData, const Shape& scoresShape, in boxWithNmsLimitFloat16()
643 const _Float16* roiData, const Shape& roiShape, in boxWithNmsLimitFloat16()
644 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat16()
648 const Shape& scoresOutShape, _Float16* roiOutData, in boxWithNmsLimitFloat16()
649 const Shape& roiOutShape, int32_t* classesOutData, in boxWithNmsLimitFloat16()
650 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat16()
651 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitFloat16()
665 bool boxWithNmsLimitQuant(const uint8_t* scoresData, const Shape& scoresShape, in boxWithNmsLimitQuant()
666 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant()
667 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant()
670 uint8_t* scoresOutData, const Shape& scoresOutShape, uint16_t* roiOutData, in boxWithNmsLimitQuant()
671 const Shape& roiOutShape, int32_t* classesOutData, in boxWithNmsLimitQuant()
672 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()
673 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitQuant()
687 bool boxWithNmsLimitQuant(const int8_t* scoresData, const Shape& scoresShape, in boxWithNmsLimitQuant()
688 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant()
689 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant()
692 int8_t* scoresOutData, const Shape& scoresOutShape, uint16_t* roiOutData, in boxWithNmsLimitQuant()
693 const Shape& roiOutShape, int32_t* classesOutData, in boxWithNmsLimitQuant()
694 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()
695 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitQuant()
760 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
761 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
762 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
763 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
764 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
765 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
766 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
959 bool generateProposalsNhwcFloat32Compute(const float* scoresData, const Shape& scoresShape, in generateProposalsNhwcFloat32Compute()
960 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsNhwcFloat32Compute()
961 const float* anchorsData, const Shape& anchorsShape, in generateProposalsNhwcFloat32Compute()
962 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsNhwcFloat32Compute()
1003 Shape tempRoiShape = anchorsShape; in generateProposalsNhwcFloat32Compute()
1005 Shape tempBBoxDeltasShape = bboxDeltasShape; in generateProposalsNhwcFloat32Compute()
1008 Shape tempbatchSplitShape = {.dimensions = {batchSize}}; in generateProposalsNhwcFloat32Compute()
1009 Shape tempImageInfoShape = imageInfoShape; in generateProposalsNhwcFloat32Compute()
1061 bool generateProposalsFloat32Compute(const float* scoresData, const Shape& scoresShape, in generateProposalsFloat32Compute()
1062 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat32Compute()
1063 const float* anchorsData, const Shape& anchorsShape, in generateProposalsFloat32Compute()
1064 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat32Compute()
1080 bool generateProposalsFloat32(const float* scoresData, const Shape& scoresShape, in generateProposalsFloat32()
1081 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat32()
1082 const float* anchorsData, const Shape& anchorsShape, in generateProposalsFloat32()
1083 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat32()
1097 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat32()
1100 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat32()
1103 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat32()
1123 bool generateProposalsFloat16(const _Float16* scoresData, const Shape& scoresShape, in generateProposalsFloat16()
1124 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat16()
1125 const _Float16* anchorsData, const Shape& anchorsShape, in generateProposalsFloat16()
1126 const _Float16* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat16()
1149 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat16()
1152 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat16()
1155 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat16()
1172 bool generateProposalsQuant(const T_8QInput* scoresData, const Shape& scoresShape, in generateProposalsQuant()
1173 const T_8QInput* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsQuant()
1174 const int16_t* anchorsData, const Shape& anchorsShape, in generateProposalsQuant()
1175 const uint16_t* imageInfoData, const Shape& imageInfoShape, in generateProposalsQuant()
1201 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsQuant()
1204 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsQuant()
1207 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsQuant()
1291 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1292 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
1293 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1294 Shape imageInfoDataShape = context->getInputShape(kImageInfoTensor); in prepare()
1295 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1296 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1297 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
1452 const float* scoreData, const Shape& scoreShape, const float* deltaData, in detectionPostprocessFloat32()
1453 const Shape& deltaShape, const float* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat32()
1456 float scoreThreshold, bool isBGInLabel, float* scoreOutData, const Shape& scoreOutShape, in detectionPostprocessFloat32()
1457 float* roiOutData, const Shape& roiOutShape, int32_t* classOutData, in detectionPostprocessFloat32()
1458 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat32()
1557 const _Float16* scoreData, const Shape& scoreShape, const _Float16* deltaData, in detectionPostprocessFloat16()
1558 const Shape& deltaShape, const _Float16* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat16()
1561 float scoreThreshold, bool isBGInLabel, _Float16* scoreOutData, const Shape& scoreOutShape, in detectionPostprocessFloat16()
1562 _Float16* roiOutData, const Shape& roiOutShape, int32_t* classOutData, in detectionPostprocessFloat16()
1563 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat16()
1620 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1621 Shape deltasShape = context->getInputShape(kDeltaTensor); in prepare()
1622 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1623 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1624 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1625 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
1626 Shape outputDetectionShape = context->getOutputShape(kOutputDetectionTensor); in prepare()