| 55 | } |
| 56 | |
| 57 | size_t detectionForwardPreNMSSize(int32_t N, int32_t C2) |
| 58 | { |
| 59 | PLUGIN_ASSERT(sizeof(float) == sizeof(int32_t)); |
| 60 | return N * C2 * sizeof(float); |
| 61 | } |
| 62 | |
| 63 | size_t detectionForwardPostNMSSize(int32_t N, int32_t numClasses, int32_t topK) |
| 64 | { |
no outgoing calls
no test coverage detected