TensorRT 8.6.1
NvInferPlugin.h
Go to the documentation of this file.
1/*
2 * SPDX-FileCopyrightText: Copyright (c) 1993-2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
4 *
5 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
6 * property and proprietary rights in and to this material, related
7 * documentation and any modifications thereto. Any use, reproduction,
8 * disclosure or distribution of this material and related documentation
9 * without an express license agreement from NVIDIA CORPORATION or
10 * its affiliates is strictly prohibited.
11 */
12
13#ifndef NV_INFER_PLUGIN_H
14#define NV_INFER_PLUGIN_H
15
16#include "NvInfer.h"
17#include "NvInferPluginUtils.h"
23
24extern "C"
25{
44 TRT_DEPRECATED_API nvinfer1::IPluginV2* createRPNROIPlugin(int32_t featureStride, int32_t preNmsTop, int32_t nmsMaxOut,
45 float iouThreshold, float minBoxSize, float spatialScale, nvinfer1::DimsHW pooling,
46 nvinfer1::Weights anchorRatios, nvinfer1::Weights anchorScales);
47
61 nvinfer1::Weights const* scales, bool acrossSpatial, bool channelShared, float eps);
62
72
83 nvinfer1::plugin::GridAnchorParameters* param, int32_t numLayers);
84
94
106
118
152
164 TRT_DEPRECATED_API nvinfer1::IPluginV2* createSplitPlugin(int32_t axis, int32_t* output_lengths, int32_t noutput);
165
179 float epsilon, nvinfer1::Weights scale_weights, nvinfer1::Weights bias_weights);
180
188 TENSORRTAPI bool initLibNvInferPlugins(void* logger, char const* libNamespace);
189
190} // extern "C"
191
192#endif // NV_INFER_PLUGIN_H
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createNormalizePlugin(nvinfer1::Weights const *scales, bool acrossSpatial, bool channelShared, float eps)
The Normalize plugin layer normalizes the input to have L2 norm of 1 with scale learnable....
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createRegionPlugin(nvinfer1::plugin::RegionParameters params)
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createAnchorGeneratorPlugin(nvinfer1::plugin::GridAnchorParameters *param, int32_t numLayers)
The Grid Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratio...
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createPriorBoxPlugin(nvinfer1::plugin::PriorBoxParameters param)
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createSplitPlugin(int32_t axis, int32_t *output_lengths, int32_t noutput)
The Split Plugin performs a split operation on the input tensor. It splits the input tensor into seve...
bool initLibNvInferPlugins(void *logger, char const *libNamespace)
Initialize and register all the existing TensorRT plugins to the Plugin Registry with an optional nam...
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createReorgPlugin(int32_t stride)
The Reorg plugin reshapes input of shape CxHxW into a (C*stride*stride)x(H/stride)x(W/stride) shape,...
TRT_DEPRECATED_API 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)
Create a plugin layer that fuses the RPN and ROI pooling using user-defined parameters....
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createNMSPlugin(nvinfer1::plugin::DetectionOutputParameters param)
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createInstanceNormalizationPlugin(float epsilon, nvinfer1::Weights scale_weights, nvinfer1::Weights bias_weights)
The Instance Normalization Plugin computes the instance normalization of an input tensor....
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createBatchedNMSPlugin(nvinfer1::plugin::NMSParameters param)
The BatchedNMS Plugin performs non_max_suppression on the input boxes, per batch, across all classes....
#define TRT_DEPRECATED_API
Definition: NvInferRuntimeBase.h:42
#define TENSORRTAPI
Definition: NvInferRuntimeBase.h:54
Descriptor for two-dimensional spatial data.
Definition: NvInferLegacyDims.h:70
Plugin class for user-implemented layers.
Definition: NvInferRuntimePlugin.h:98
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:126
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPluginUtils.h:154
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPluginUtils.h:115
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPluginUtils.h:218
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPluginUtils.h:61
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPluginUtils.h:193

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact