|
TensorRT
|
| The DetectionOutput plugin layer generates the detection output based on location and confidence predictions by doing non maximum suppression. DetectionOutputParameters defines a set of parameters for creating the DetectionOutput plugin layer. It contains: | |
| Structure to define the dimensions of a tensor | |
| Descriptor for two-dimensional data | |
| Descriptor for two-dimensional spatial data | |
| Descriptor for three-dimensional data | |
| Descriptor for data with one channel dimension and two spatial dimensions | |
| Descriptor for four-dimensional data | |
| Descriptor for data with one index dimension, one channel dimension and two spatial dimensions | |
| An array of field params used as a layer parameter for plugin layers | |
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios across all dimensions . GridAnchorParameters defines a set of parameters for creating the plugin layer for all feature maps. It contains: | |
| Object used to store and query data extracted from a binaryproto file using the ICaffeParser | |
| Object used to store and query Tensors after they have been extracted from a Caffe model using the ICaffeParser | |
| Builds an engine from a network definition | |
| Class used for parsing Caffe models | |
| An engine for executing inference on a built network | |
| Context for executing inference using an engine | |
| Application-implemented class for controlling allocation on the GPU | |
| Class to handle library allocated memory that is accessible to the user | |
| Application-implemented interface for calibration | |
| Base class for all layer classes in a network definition | |
| An Activation layer in a network definition | |
| A concatenation layer in a network definition | |
| Layer that represents a constant value | |
| A convolution layer in a network definition | |
| A deconvolution layer in a network definition | |
| A elementwise layer in a network definition | |
A fully connected layer in a network definition. This layer expects an input tensor of three or more non-batch dimensions. The input is automatically reshaped into an MxV tensor X, where V is a product of the last three dimensions and M is a product of the remaining dimensions (where the product over 0 dimensions is defined as 1). For example: | |
| A LRN layer in a network definition | |
| Layer that represents a Matrix Multiplication | |
| Layer that represents a padding operation | |
| Layer type for plugins | |
| A Pooling layer in a network definition | |
| A RaggedSoftmax layer in a network definition | |
| Layer that represents a reduction operator | |
| A RNN layer in a network definition | |
| An RNN layer in a network definition, version 2 | |
| A Scale layer in a network definition | |
| Layer type for shuffling data | |
| A Softmax layer in a network definition | |
| Layer that represents a TopK reduction | |
| Layer that represents an unary operation | |
| Application-implemented logging interface for the builder, engine and runtime | |
| A network definition for input to the builder | |
| Configuration Manager Class | |
| ONNX Parser Class | |
| Application-implemented interface to compute layer output sizes | |
| Plugin class for user-implemented layers | |
| Plugin class for user-implemented layers | |
| Common interface for the Nvidia created plugins | |
| Plugin factory used to configure plugins | |
| Plugin factory used to configure plugins with added support for TRT versioning | |
| Plugin factory used to configure plugins | |
| Plugin factory used to configure plugins with added support for TRT versioning | |
| Plugin factory for deserialization | |
| Application-implemented interface for profiling | |
| Allows a serialized engine to be deserialized | |
| A tensor in a network definition | |
| Class used for parsing models described using the UFF format | |
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all dimensions . PriorBoxParameters defines a set of parameters for creating the PriorBox plugin layer. It contains: | |
| The Permute plugin layer permutes the input tensor by changing the memory order of the data. Quadruple defines a structure that contains an array of 4 integers. They can represent the permute orders or the strides in each dimension | |
| The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell). For each box, calculating its probablities of objects detections from 80 pre-defined classifications (yolo9000 has 9416 pre-defined classifications, and these 9416 items are organized as work-tree structure). RegionParameters defines a set of parameters for creating the Region plugin layer | |
| An array of weights used as a layer parameter |