NvInferPlugin.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef NV_INFER_PLUGIN_H
#define NV_INFER_PLUGIN_H
 
#include "NvInfer.h"
#include "NvInferPluginUtils.h"
 
extern "C"
{
    TENSORRTAPI nvinfer1::IPluginV2* createRPNROIPlugin(int32_t featureStride, int32_t preNmsTop, int32_t nmsMaxOut,
        float iouThreshold, float minBoxSize, float spatialScale, nvinfer1::DimsHW pooling,
        nvinfer1::Weights anchorRatios, nvinfer1::Weights anchorScales);
 
    TENSORRTAPI nvinfer1::IPluginV2* createNormalizePlugin(
        nvinfer1::Weights const* scales, bool acrossSpatial, bool channelShared, float eps);
 
    TENSORRTAPI nvinfer1::IPluginV2* createPriorBoxPlugin(nvinfer1::plugin::PriorBoxParameters param);
 
    TENSORRTAPI nvinfer1::IPluginV2* createAnchorGeneratorPlugin(
        nvinfer1::plugin::GridAnchorParameters* param, int32_t numLayers);
 
    TENSORRTAPI nvinfer1::IPluginV2* createNMSPlugin(nvinfer1::plugin::DetectionOutputParameters param);
 
    TENSORRTAPI nvinfer1::IPluginV2* createReorgPlugin(int32_t stride);
 
    TENSORRTAPI nvinfer1::IPluginV2* createRegionPlugin(nvinfer1::plugin::RegionParameters params);
 
    TENSORRTAPI nvinfer1::IPluginV2* createBatchedNMSPlugin(nvinfer1::plugin::NMSParameters param);
 
    TENSORRTAPI nvinfer1::IPluginV2* createSplitPlugin(int32_t axis, int32_t* output_lengths, int32_t noutput);
 
    TENSORRTAPI nvinfer1::IPluginV2* createInstanceNormalizationPlugin(
        float epsilon, nvinfer1::Weights scale_weights, nvinfer1::Weights bias_weights);
 
    TENSORRTAPI bool initLibNvInferPlugins(void* logger, char const* libNamespace);
 
} // extern "C"
 
#endif // NV_INFER_PLUGIN_H