Lines Matching refs:Shape

52 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape,  in bboxTransformFloat32()
53 const float* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat32()
54 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat32()
55 const float* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat32()
56 float* outputData, const Shape& outputShape) { in bboxTransformFloat32()
102 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16()
103 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformFloat16()
104 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16()
105 const _Float16* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformFloat16()
106 _Float16* outputData, const Shape& outputShape) { in bboxTransformFloat16()
122 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant()
123 const uint8_t* bboxDeltasData, const Shape& bboxDeltasShape, in bboxTransformQuant()
124 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant()
125 const uint16_t* imageInfoData, const Shape& imageInfoDataShape, in bboxTransformQuant()
126 uint16_t* outputData, const Shape& outputShape) { in bboxTransformQuant()
194 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
195 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
196 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
197 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
198 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
433 bool boxWithNmsLimitFloat32Compute(float* scoresData, const Shape& scoresShape, in boxWithNmsLimitFloat32Compute()
434 const float* roiData, const Shape& roiShape, in boxWithNmsLimitFloat32Compute()
435 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat32Compute()
498 T castTo(float val, const Shape&) { in castTo() argument
502 uint8_t castTo(float val, const Shape& shape) { in castTo()
516 Shape scoresShape = context->getInputShape(kScoreTensor); in boxWithNmsLimitWriteOutput()
522 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
526 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
530 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
534 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
563 bool boxWithNmsLimitFloat32(const float* scoresData, const Shape& scoresShape, const float* roiData, in boxWithNmsLimitFloat32()
564 const Shape& roiShape, const int32_t* batchesData, in boxWithNmsLimitFloat32()
565 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32()
568 Shape scoresOutShape, float* roiOutData, Shape roiOutShape, in boxWithNmsLimitFloat32()
569 int32_t* classesOutData, Shape classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat32()
570 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitFloat32()
585 bool boxWithNmsLimitFloat16(const _Float16* scoresData, const Shape& scoresShape, in boxWithNmsLimitFloat16()
586 const _Float16* roiData, const Shape& roiShape, in boxWithNmsLimitFloat16()
587 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat16()
591 const Shape& scoresOutShape, _Float16* roiOutData, in boxWithNmsLimitFloat16()
592 const Shape& roiOutShape, int32_t* classesOutData, in boxWithNmsLimitFloat16()
593 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat16()
594 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitFloat16()
608 bool boxWithNmsLimitQuant(const uint8_t* scoresData, const Shape& scoresShape, in boxWithNmsLimitQuant()
609 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant()
610 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant()
613 uint8_t* scoresOutData, const Shape& scoresOutShape, uint16_t* roiOutData, in boxWithNmsLimitQuant()
614 const Shape& roiOutShape, int32_t* classesOutData, in boxWithNmsLimitQuant()
615 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()
616 const Shape& batchSplitOutShape, IOperationExecutionContext* context) { in boxWithNmsLimitQuant()
673 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
674 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
675 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
676 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
677 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
678 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
679 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
843 bool generateProposalsNhwcFloat32Compute(const float* scoresData, const Shape& scoresShape, in generateProposalsNhwcFloat32Compute()
844 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsNhwcFloat32Compute()
845 const float* anchorsData, const Shape& anchorsShape, in generateProposalsNhwcFloat32Compute()
846 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsNhwcFloat32Compute()
887 Shape tempRoiShape = anchorsShape; in generateProposalsNhwcFloat32Compute()
889 Shape tempBBoxDeltasShape = bboxDeltasShape; in generateProposalsNhwcFloat32Compute()
892 Shape tempbatchSplitShape = {.dimensions = {batchSize}}; in generateProposalsNhwcFloat32Compute()
893 Shape tempImageInfoShape = imageInfoShape; in generateProposalsNhwcFloat32Compute()
945 bool generateProposalsFloat32Compute(const float* scoresData, const Shape& scoresShape, in generateProposalsFloat32Compute()
946 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat32Compute()
947 const float* anchorsData, const Shape& anchorsShape, in generateProposalsFloat32Compute()
948 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat32Compute()
964 bool generateProposalsFloat32(const float* scoresData, const Shape& scoresShape, in generateProposalsFloat32()
965 const float* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat32()
966 const float* anchorsData, const Shape& anchorsShape, in generateProposalsFloat32()
967 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat32()
981 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat32()
984 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat32()
987 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat32()
1007 bool generateProposalsFloat16(const _Float16* scoresData, const Shape& scoresShape, in generateProposalsFloat16()
1008 const _Float16* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsFloat16()
1009 const _Float16* anchorsData, const Shape& anchorsShape, in generateProposalsFloat16()
1010 const _Float16* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat16()
1033 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat16()
1036 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat16()
1039 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat16()
1055 bool generateProposalsQuant(const uint8_t* scoresData, const Shape& scoresShape, in generateProposalsQuant()
1056 const uint8_t* bboxDeltasData, const Shape& bboxDeltasShape, in generateProposalsQuant()
1057 const int16_t* anchorsData, const Shape& anchorsShape, in generateProposalsQuant()
1058 const uint16_t* imageInfoData, const Shape& imageInfoShape, in generateProposalsQuant()
1083 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsQuant()
1086 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsQuant()
1089 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsQuant()
1166 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1167 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
1168 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1169 Shape imageInfoDataShape = context->getInputShape(kImageInfoTensor); in prepare()
1170 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1171 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1172 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
1308 const float* scoreData, const Shape& scoreShape, const float* deltaData, in detectionPostprocessFloat32()
1309 const Shape& deltaShape, const float* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat32()
1312 float scoreThreshold, bool isBGInLabel, float* scoreOutData, const Shape& scoreOutShape, in detectionPostprocessFloat32()
1313 float* roiOutData, const Shape& roiOutShape, int32_t* classOutData, in detectionPostprocessFloat32()
1314 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat32()
1413 const _Float16* scoreData, const Shape& scoreShape, const _Float16* deltaData, in detectionPostprocessFloat16()
1414 const Shape& deltaShape, const _Float16* anchorData, const Shape& anchorShape, float scaleY, in detectionPostprocessFloat16()
1417 float scoreThreshold, bool isBGInLabel, _Float16* scoreOutData, const Shape& scoreOutShape, in detectionPostprocessFloat16()
1418 _Float16* roiOutData, const Shape& roiOutShape, int32_t* classOutData, in detectionPostprocessFloat16()
1419 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat16()
1474 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1475 Shape deltasShape = context->getInputShape(kDeltaTensor); in prepare()
1476 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1477 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1478 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1479 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
1480 Shape outputDetectionShape = context->getOutputShape(kOutputDetectionTensor); in prepare()