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

Method serialize

plugin/flattenConcat/flattenConcat.cpp:192–210  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

190}
191
192void FlattenConcat::serialize(void* buffer) const noexcept
193{
194 char* d = static_cast<char*>(buffer);
195 char const* const a = d;
196 write(d, mIgnoreBatch);
197 write(d, mConcatAxisID);
198 write(d, mOutputConcatAxis);
199 write(d, mNumInputs);
200 for (int32_t i = 0; i < mNumInputs; ++i)
201 {
202 write(d, mInputConcatAxis[i]);
203 }
204 write(d, mCHW);
205 for (int32_t i = 0; i < mNumInputs; ++i)
206 {
207 write(d, mCopySize[i]);
208 }
209 PLUGIN_ASSERT(d == a + getSerializationSize());
210}
211
212// Attach the plugin object to an execution context and grant the plugin the access to some context resource.
213void FlattenConcat::attachToContext(

Callers

nothing calls this directly

Calls 1

writeFunction · 0.85

Tested by

no test coverage detected