Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Class and Description |
---|---|
class |
IActivationLayer
\class IActivationLayer
\brief An Activation layer in a network definition.
|
class |
IAssertionLayer
\class IAssertionLayer
\brief An assertion layer in a network
The layer has a single input and no output.
|
class |
ICastLayer
\class ICastLayer
\brief A cast layer in a network.
|
class |
IConcatenationLayer
\class IConcatenationLayer
\brief A concatenation layer in a network definition.
|
class |
IConditionLayer
This layer represents a condition input to an IIfConditional.
|
class |
IConstantLayer
\class IConstantLayer
\brief Layer that represents a constant value.
|
class |
IConvolutionLayer
\class IConvolutionLayer
\brief A convolution layer in a network definition.
|
class |
IDeconvolutionLayer
\class IDeconvolutionLayer
\brief A deconvolution layer in a network definition.
|
class |
IDequantizeLayer
\class IDequantizeLayer
\brief A Dequantize layer in a network definition.
|
class |
IEinsumLayer
\class IEinsumLayer
\brief An Einsum layer in a network
This layer implements a summation over the elements of the inputs along dimensions specified by the equation
parameter, based on the Einstein summation convention.
|
class |
IElementWiseLayer
\class IElementWiseLayer
\brief A elementwise layer in a network definition.
|
class |
IFillLayer
\brief Generate an output tensor with specified mode.
|
class |
IFullyConnectedLayer
Deprecated.
Deprecated in TensorRT 8.4. Superseded by IMatrixMultiplyLayer.
|
class |
IGatherLayer
\class IGatherLayer
\brief A Gather layer in a network definition.
|
class |
IGridSampleLayer
\class IGridSampleLayer
\brief A GridSample layer in a network definition.
|
class |
IIdentityLayer
\class IIdentityLayer
\brief A layer that represents the identity function.
|
class |
IIfConditionalBoundaryLayer
This is a base class for Conditional boundary layers.
|
class |
IIfConditionalInputLayer
This layer represents an input to an IIfConditional.
|
class |
IIfConditionalOutputLayer
This layer represents an output of an IIfConditional.
|
class |
IIteratorLayer |
class |
ILoopBoundaryLayer |
class |
ILoopOutputLayer
An ILoopOutputLayer is the sole way to get output from a loop.
|
class |
ILRNLayer
\class ILRNLayer
\brief A LRN layer in a network definition.
|
class |
IMatrixMultiplyLayer
\class IMatrixMultiplyLayer
\brief Layer that represents a Matrix Multiplication.
|
class |
INMSLayer
\class INMSLayer
\brief A non-maximum suppression layer in a network definition.
|
class |
INonZeroLayer
\class INonZero
\brief A NonZero layer in a network.
|
class |
INormalizationLayer
\class INormalizationLayer
\brief A normalization layer in a network definition.
|
class |
IOneHotLayer
\class IOneHotLayer
\brief A OneHot layer in a network definition.
|
class |
IPaddingLayer
\class IPaddingLayer
\brief Layer that represents a padding operation.
|
class |
IParametricReLULayer
\class IParametricReLULayer
\brief Layer that represents a parametric ReLU operation.
|
class |
IPluginV2Layer
\class IPluginV2Layer
\brief Layer type for pluginV2
|
class |
IPoolingLayer
\class IPoolingLayer
\brief A Pooling layer in a network definition.
|
class |
IQuantizeLayer
\class IQuantizeLayer
\brief A Quantize layer in a network definition.
|
class |
IRaggedSoftMaxLayer
\class IRaggedSoftMaxLayer
\brief A RaggedSoftmax layer in a network definition.
|
class |
IRecurrenceLayer |
class |
IReduceLayer
\class IReduceLayer
\brief Layer that represents a reduction across a non-bool tensor.
|
class |
IResizeLayer
\class IResizeLayer
\brief A resize layer in a network definition.
|
class |
IReverseSequenceLayer
\class IReverseSequenceLayer
\brief A ReverseSequence layer in a network definition.
|
class |
IRNNv2Layer
Deprecated.
Deprecated prior to TensorRT 8.0 and will be removed in 9.0. Superseded by
INetworkDefinition::addLoop().
\warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
|
class |
IScaleLayer
\class IScaleLayer
\brief A Scale layer in a network definition.
|
class |
IScatterLayer
\class IScatterLayer
\brief A scatter layer in a network definition.
|
class |
ISelectLayer
\warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
|
class |
IShapeLayer
\class IShapeLayer
\brief Layer type for getting shape of a tensor.
|
class |
IShuffleLayer
\class IShuffleLayer
\brief Layer type for shuffling data.
|
class |
ISliceLayer
\brief Slices an input tensor into an output tensor based on the offset and strides.
|
class |
ISoftMaxLayer
\class ISoftMaxLayer
\brief A Softmax layer in a network definition.
|
class |
ITopKLayer
\class ITopKLayer
\brief Layer that represents a TopK reduction.
|
class |
ITripLimitLayer |
class |
IUnaryLayer
\class IUnaryLayer
\brief Layer that represents an unary operation.
|
Modifier and Type | Method and Description |
---|---|
ILayer |
INetworkDefinition.getLayer(int index)
\brief Get the layer specified by the given index.
|
ILayer |
VNetworkDefinition.getLayer(int index) |
Modifier and Type | Method and Description |
---|---|
boolean |
IBuilderConfig.canRunOnDLA(ILayer layer)
\brief Checks if a layer can run on DLA.
|
boolean |
VBuilderConfig.canRunOnDLA(ILayer layer) |
nvinfer.DeviceType |
IBuilderConfig.getDeviceType(ILayer layer)
\brief Get the device that this layer executes on.
|
nvinfer.DeviceType |
VBuilderConfig.getDeviceType(ILayer layer) |
boolean |
IBuilderConfig.isDeviceTypeSet(ILayer layer)
\brief whether the DeviceType has been explicitly set for this layer
|
boolean |
VBuilderConfig.isDeviceTypeSet(ILayer layer) |
void |
IBuilderConfig.resetDeviceType(ILayer layer)
\brief reset the DeviceType for this layer
|
void |
VBuilderConfig.resetDeviceType(ILayer layer) |
void |
IBuilderConfig.setDeviceType(ILayer layer,
int deviceType) |
void |
VBuilderConfig.setDeviceType(ILayer layer,
int deviceType) |
void |
IBuilderConfig.setDeviceType(ILayer layer,
nvinfer.DeviceType deviceType)
\brief Set the device that this layer must execute on.
|
void |
VBuilderConfig.setDeviceType(ILayer layer,
nvinfer.DeviceType deviceType) |
Copyright © 2024. All rights reserved.