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

Method serialize

plugin/normalizePlugin/normalizePlugin.cpp:140–154  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

138}
139
140void Normalize::serialize(void* buffer) const noexcept
141{
142 char *d = static_cast<char*>(buffer), *a = d;
143 write(d, C);
144 write(d, H);
145 write(d, W);
146 write(d, acrossSpatial);
147 write(d, channelShared);
148 write(d, eps);
149 write(d, (int32_t) mNbWeights);
150 write(d, (int32_t) mWeights.count);
151 serializeFromDevice(d, mWeights);
152
153 PLUGIN_ASSERT(d == a + getSerializationSize());
154}
155
156bool Normalize::supportsFormat(DataType type, PluginFormat format) const noexcept
157{

Callers

nothing calls this directly

Calls 1

writeFunction · 0.85

Tested by

no test coverage detected