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

Method FlattenConcatPluginCreator

plugin/flattenConcat/flattenConcat.cpp:341–349  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

339}
340
341FlattenConcatPluginCreator::FlattenConcatPluginCreator()
342{
343 mPluginAttributes.clear();
344 mPluginAttributes.emplace_back(PluginField("axis", nullptr, PluginFieldType::kINT32, 1));
345 mPluginAttributes.emplace_back(PluginField("ignoreBatch", nullptr, PluginFieldType::kINT32, 1));
346
347 mFC.nbFields = mPluginAttributes.size();
348 mFC.fields = mPluginAttributes.data();
349}
350
351char const* FlattenConcatPluginCreator::getPluginName() const noexcept
352{

Callers

nothing calls this directly

Calls 4

PluginFieldClass · 0.85
clearMethod · 0.45
sizeMethod · 0.45
dataMethod · 0.45

Tested by

no test coverage detected