| 243 | } |
| 244 | |
| 245 | Dims ProposalPlugin::getOutputDimensions(int32_t index, Dims const* inputs, int32_t nbInputDims) noexcept |
| 246 | { |
| 247 | try |
| 248 | { |
| 249 | // Validate input arguments |
| 250 | PLUGIN_VALIDATE(index == 0); |
| 251 | PLUGIN_VALIDATE(nbInputDims == 2); |
| 252 | PLUGIN_VALIDATE(inputs->nbDims == 3); |
| 253 | PLUGIN_VALIDATE(inputs[1].nbDims == 3); |
| 254 | int32_t channels = mMaxBoxNum; |
| 255 | int32_t height = 4; |
| 256 | int32_t width = 1; |
| 257 | return Dims3(channels, height, width); |
| 258 | } |
| 259 | catch (std::exception const& e) |
| 260 | { |
| 261 | caughtError(e); |
| 262 | } |
| 263 | return Dims{}; |
| 264 | } |
| 265 | |
| 266 | DimsExprs ProposalDynamicPlugin::getOutputDimensions( |
| 267 | int32_t outputIndex, DimsExprs const* inputs, int32_t nbInputs, IExprBuilder& exprBuilder) noexcept |
nothing calls this directly
no test coverage detected