| 319 | } |
| 320 | |
| 321 | void BatchedNMSPlugin::serialize(void* buffer) const noexcept |
| 322 | { |
| 323 | char *d = reinterpret_cast<char*>(buffer), *a = d; |
| 324 | write(d, param); |
| 325 | write(d, mBoxesSize); |
| 326 | write(d, mScoresSize); |
| 327 | write(d, mNumPriors); |
| 328 | write(d, mClipBoxes); |
| 329 | write(d, mPrecision); |
| 330 | write(d, mScoreBits); |
| 331 | write(d, mCaffeSemantics); |
| 332 | PLUGIN_ASSERT(d == a + getSerializationSize()); |
| 333 | } |
| 334 | |
| 335 | size_t BatchedNMSDynamicPlugin::getSerializationSize() const noexcept |
| 336 | { |