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

Method getOutputDimensions

plugin/batchTilePlugin/batchTilePlugin.cpp:61–76  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

59}
60
61Dims BatchTilePlugin::getOutputDimensions(int32_t index, Dims const* inputs, int32_t nbInputDims) noexcept
62{
63 try
64 {
65 PLUGIN_VALIDATE(inputs != nullptr);
66 PLUGIN_VALIDATE(nbInputDims == 2);
67 PLUGIN_VALIDATE(index == 0);
68 PLUGIN_VALIDATE(inputs[1].nbDims == 4);
69 return Dims3(inputs[1].d[1], inputs[1].d[2], inputs[1].d[3]);
70 }
71 catch (std::exception const& e)
72 {
73 caughtError(e);
74 }
75 return Dims{};
76}
77
78int32_t BatchTilePlugin::initialize() noexcept
79{

Callers

nothing calls this directly

Calls 2

Dims3Class · 0.85
caughtErrorFunction · 0.85

Tested by

no test coverage detected