Function detectionForwardPostNMSSize

plugin/common/nmsHelper.cpp:63–67

}

size_t detectionForwardPostNMSSize(int32_t N, int32_t numClasses, int32_t topK)
{
    PLUGIN_ASSERT(sizeof(float) == sizeof(int32_t));
    return N * numClasses * topK * sizeof(float);
}
} // namespace plugin
} // namespace nvinfer1