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

Method FCPluginDynamic

plugin/fcPlugin/fcPlugin.cpp:369–382  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

367}
368
369FCPluginDynamic::FCPluginDynamic(std::string const name, DataType const type, int32_t const outDim, Weights const& W)
370 : mLayerName(name)
371 , mType(type)
372 , mOutDim(outDim)
373 , mNumParams(W.count)
374 , mNmax(0)
375 , mK(0)
376 , mWdev(nullptr)
377{
378 memset(mAlgo.data, 0, sizeof(mAlgo.data));
379
380 mW.convertAndCopy(W, mType);
381 copyToDevice(mW, getWeightsSize(mW, mType), mWdev);
382}
383
384FCPluginDynamic::FCPluginDynamic(std::string const name, void const* data, size_t length)
385 : mLayerName(name)

Callers

nothing calls this directly

Calls 4

copyToDeviceFunction · 0.85
getWeightsSizeFunction · 0.85
deserialize_valueFunction · 0.85
convertAndCopyMethod · 0.80

Tested by

no test coverage detected