| 40 | } |
| 41 | |
| 42 | size_t detectionForwardBBoxPermuteSize(bool shareLocation, int32_t N, int32_t C1, DataType DT_BBOX) |
| 43 | { |
| 44 | if (DT_BBOX == DataType::kFLOAT) |
| 45 | { |
| 46 | return shareLocation ? 0 : N * C1 * sizeof(float); |
| 47 | } |
| 48 | if (DT_BBOX == DataType::kHALF) |
| 49 | { |
| 50 | return shareLocation ? 0 : N * C1 * sizeof(__half); |
| 51 | } |
| 52 | |
| 53 | printf("Only FP32/FP16 type bounding boxes are supported.\n"); |
| 54 | return (size_t) -1; |
| 55 | } |
| 56 | |
| 57 | size_t detectionForwardPreNMSSize(int32_t N, int32_t C2) |
| 58 | { |
no outgoing calls
no test coverage detected