| 274 | } |
| 275 | |
| 276 | Dims DetectionLayer::getOutputDimensions(int32_t index, Dims const* inputs, int32_t nbInputDims) noexcept |
| 277 | { |
| 278 | try |
| 279 | { |
| 280 | checkValidInputs(inputs, nbInputDims); |
| 281 | PLUGIN_VALIDATE(index == 0); |
| 282 | // [N, anchors, (y1, x1, y2, x2, class_id, score)] |
| 283 | return {2, {mKeepTopK, 6}}; |
| 284 | } |
| 285 | catch (std::exception const& e) |
| 286 | { |
| 287 | caughtError(e); |
| 288 | } |
| 289 | return Dims{}; |
| 290 | } |
| 291 | |
| 292 | int32_t DetectionLayer::enqueue( |
| 293 | int32_t batchSize, void const* const* inputs, void* const* outputs, void* workspace, cudaStream_t stream) noexcept |
nothing calls this directly
no test coverage detected