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

Method getWorkspaceSize

plugin/nmsPlugin/nmsPlugin.cpp:211–215  ·  view source on GitHub ↗

Returns the workspace size

Source from the content-addressed store, hash-verified

209
210// Returns the workspace size
211size_t DetectionOutput::getWorkspaceSize(int32_t maxBatchSize) const noexcept
212{
213 return detectionInferenceWorkspaceSize(
214 param.shareLocation, maxBatchSize, C1, C2, param.numClasses, numPriors, param.topK, mType, mType);
215}
216
217size_t DetectionOutputDynamic::getWorkspaceSize(
218 PluginTensorDesc const* inputs, int32_t nbInputs, PluginTensorDesc const* outputs, int32_t nbOutputs) const noexcept

Callers

nothing calls this directly

Calls 1

Tested by

no test coverage detected