| 25 | { |
| 26 | |
| 27 | size_t detectionForwardBBoxDataSize(int32_t N, int32_t C1, DataType DT_BBOX) |
| 28 | { |
| 29 | if (DT_BBOX == DataType::kFLOAT) |
| 30 | { |
| 31 | return N * C1 * sizeof(float); |
| 32 | } |
| 33 | if (DT_BBOX == DataType::kHALF) |
| 34 | { |
| 35 | return N * C1 * sizeof(__half); |
| 36 | } |
| 37 | |
| 38 | printf("Only FP32/FP16 type bounding boxes are supported.\n"); |
| 39 | return (size_t) -1; |
| 40 | } |
| 41 | |
| 42 | size_t detectionForwardBBoxPermuteSize(bool shareLocation, int32_t N, int32_t C1, DataType DT_BBOX) |
| 43 | { |
no outgoing calls
no test coverage detected