| 232 | } |
| 233 | |
| 234 | void GridAnchorGenerator::serialize(void* buffer) const noexcept |
| 235 | { |
| 236 | char *d = reinterpret_cast<char*>(buffer), *a = d; |
| 237 | write(d, mNumLayers); |
| 238 | for (int32_t id = 0; id < mNumLayers; id++) |
| 239 | { |
| 240 | // we have to serialize GridAnchorParameters by hand |
| 241 | write(d, mParam[id].minSize); |
| 242 | write(d, mParam[id].maxSize); |
| 243 | write(d, mParam[id].numAspectRatios); |
| 244 | for (int32_t i = 0; i < mParam[id].numAspectRatios; ++i) |
| 245 | { |
| 246 | write(d, mParam[id].aspectRatios[i]); |
| 247 | } |
| 248 | write(d, mParam[id].H); |
| 249 | write(d, mParam[id].W); |
| 250 | for (int32_t i = 0; i < 4; ++i) |
| 251 | { |
| 252 | write(d, mParam[id].variance[i]); |
| 253 | } |
| 254 | |
| 255 | write(d, mNumPriors[id]); |
| 256 | serializeFromDevice(d, mDeviceWidths[id]); |
| 257 | serializeFromDevice(d, mDeviceHeights[id]); |
| 258 | } |
| 259 | PLUGIN_ASSERT(d == a + getSerializationSize()); |
| 260 | } |
| 261 | |
| 262 | Weights GridAnchorGenerator::copyToDevice(void const* hostData, size_t count) noexcept |
| 263 | { |