MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / serialize

Method serialize

plugin/skipLayerNormPlugin/skipLayerNormPlugin.cpp:395–416  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

393}
394
395void SkipLayerNormPluginDynamic::serialize(void* buffer) const noexcept
396{
397 try
398 {
399 serialize_value(&buffer, mType);
400 serialize_value(&buffer, mCfgType);
401 serialize_value(&buffer, mLd);
402 serialize_value(&buffer, mHasBias);
403
404 char* d = static_cast<char*>(buffer);
405 serFromDev(d, static_cast<char*>(mBetaDev.get()), mLd * mParamWordsize);
406 serFromDev(d, static_cast<char*>(mGammaDev.get()), mLd * mParamWordsize);
407 if (mHasBias)
408 {
409 serFromDev(d, static_cast<char*>(mBiasDev.get()), mLd * mParamWordsize);
410 }
411 }
412 catch (std::exception const& e)
413 {
414 caughtError(e);
415 }
416}
417
418void SkipLayerNormPluginDynamic::destroy() noexcept
419{

Callers

nothing calls this directly

Calls 4

serialize_valueFunction · 0.85
serFromDevFunction · 0.85
caughtErrorFunction · 0.85
getMethod · 0.45

Tested by

no test coverage detected