| Package | Description |
|---|---|
| org.bytedeco.tensorrt.nvinfer | |
| org.bytedeco.tensorrt.nvparsers |
| Modifier and Type | Method and Description |
|---|---|
ITensor |
INetworkDefinition.addInput(BytePointer name,
int type,
Dims32 dimensions) |
ITensor |
VNetworkDefinition.addInput(BytePointer name,
int type,
Dims32 dimensions) |
ITensor |
VNetworkDefinition.addInput(BytePointer name,
int type,
Dims32 dimensions) |
ITensor |
INetworkDefinition.addInput(BytePointer name,
int type,
Dims32 dimensions) |
ITensor |
INetworkDefinition.addInput(String name,
nvinfer.DataType type,
Dims32 dimensions)
\brief Add an input tensor to the network.
|
ITensor |
VNetworkDefinition.addInput(String name,
nvinfer.DataType type,
Dims32 dimensions) |
ITensor |
VNetworkDefinition.addInput(String name,
nvinfer.DataType type,
Dims32 dimensions) |
ITensor |
INetworkDefinition.addInput(String name,
nvinfer.DataType type,
Dims32 dimensions)
\brief Add an input tensor to the network.
|
ITensor |
IRNNv2Layer.getCellState()
Deprecated.
\brief Get the initial cell state of the RNN.
|
ITensor |
VRNNv2Layer.getCellState() |
ITensor |
VRNNv2Layer.getCellState() |
ITensor |
IRNNv2Layer.getCellState()
Deprecated.
\brief Get the initial cell state of the RNN.
|
ITensor |
IRNNv2Layer.getHiddenState()
Deprecated.
\brief Get the initial hidden state of the RNN.
|
ITensor |
VRNNv2Layer.getHiddenState() |
ITensor |
VRNNv2Layer.getHiddenState() |
ITensor |
IRNNv2Layer.getHiddenState()
Deprecated.
\brief Get the initial hidden state of the RNN.
|
ITensor |
ILayer.getInput(int index)
\brief Get the layer input corresponding to the given index.
|
ITensor |
INetworkDefinition.getInput(int index)
\brief Get the input tensor specified by the given index.
|
ITensor |
VLayer.getInput(int index) |
ITensor |
VNetworkDefinition.getInput(int index) |
ITensor |
VNetworkDefinition.getInput(int index) |
ITensor |
VLayer.getInput(int index) |
ITensor |
INetworkDefinition.getInput(int index)
\brief Get the input tensor specified by the given index.
|
ITensor |
ILayer.getInput(int index)
\brief Get the layer input corresponding to the given index.
|
ITensor |
ILayer.getOutput(int index)
\brief Get the layer output corresponding to the given index.
|
ITensor |
INetworkDefinition.getOutput(int index)
\brief Get the output tensor specified by the given index.
|
ITensor |
VLayer.getOutput(int index) |
ITensor |
VNetworkDefinition.getOutput(int index) |
ITensor |
VNetworkDefinition.getOutput(int index) |
ITensor |
VLayer.getOutput(int index) |
ITensor |
INetworkDefinition.getOutput(int index)
\brief Get the output tensor specified by the given index.
|
ITensor |
ILayer.getOutput(int index)
\brief Get the layer output corresponding to the given index.
|
ITensor |
IRNNv2Layer.getSequenceLengths()
Deprecated.
\brief Get the sequence lengths specified for the RNN.
|
ITensor |
VRNNv2Layer.getSequenceLengths() |
ITensor |
VRNNv2Layer.getSequenceLengths() |
ITensor |
IRNNv2Layer.getSequenceLengths()
Deprecated.
\brief Get the sequence lengths specified for the RNN.
|
| Modifier and Type | Method and Description |
|---|---|
IActivationLayer |
INetworkDefinition.addActivation(ITensor input,
int type) |
IActivationLayer |
VNetworkDefinition.addActivation(ITensor input,
int type) |
IActivationLayer |
VNetworkDefinition.addActivation(ITensor input,
int type) |
IActivationLayer |
INetworkDefinition.addActivation(ITensor input,
int type) |
IActivationLayer |
INetworkDefinition.addActivation(ITensor input,
nvinfer.ActivationType type)
\brief Add an activation layer to the network.
|
IActivationLayer |
VNetworkDefinition.addActivation(ITensor input,
nvinfer.ActivationType type) |
IActivationLayer |
VNetworkDefinition.addActivation(ITensor input,
nvinfer.ActivationType type) |
IActivationLayer |
INetworkDefinition.addActivation(ITensor input,
nvinfer.ActivationType type)
\brief Add an activation layer to the network.
|
IAssertionLayer |
INetworkDefinition.addAssertion(ITensor condition,
BytePointer message) |
IAssertionLayer |
VNetworkDefinition.addAssertion(ITensor condition,
BytePointer message) |
IAssertionLayer |
VNetworkDefinition.addAssertion(ITensor condition,
BytePointer message) |
IAssertionLayer |
INetworkDefinition.addAssertion(ITensor condition,
BytePointer message) |
IAssertionLayer |
INetworkDefinition.addAssertion(ITensor condition,
String message)
\brief Add an assertion layer to the network.
|
IAssertionLayer |
VNetworkDefinition.addAssertion(ITensor condition,
String message) |
IAssertionLayer |
VNetworkDefinition.addAssertion(ITensor condition,
String message) |
IAssertionLayer |
INetworkDefinition.addAssertion(ITensor condition,
String message)
\brief Add an assertion layer to the network.
|
IConcatenationLayer |
INetworkDefinition.addConcatenation(ITensor inputs,
int nbInputs) |
IConcatenationLayer |
VNetworkDefinition.addConcatenation(ITensor inputs,
int nbInputs) |
IConcatenationLayer |
VNetworkDefinition.addConcatenation(ITensor inputs,
int nbInputs) |
IConcatenationLayer |
INetworkDefinition.addConcatenation(ITensor inputs,
int nbInputs) |
IConvolutionLayer |
INetworkDefinition.addConvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Superseded by addConvolutionNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IConvolutionLayer |
VNetworkDefinition.addConvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IConvolutionLayer |
VNetworkDefinition.addConvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IConvolutionLayer |
INetworkDefinition.addConvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Superseded by addConvolutionNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IConvolutionLayer |
INetworkDefinition.addConvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights)
\brief Add a multi-dimension convolution layer to the network.
|
IConvolutionLayer |
VNetworkDefinition.addConvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IConvolutionLayer |
VNetworkDefinition.addConvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IConvolutionLayer |
INetworkDefinition.addConvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights)
\brief Add a multi-dimension convolution layer to the network.
|
IDeconvolutionLayer |
INetworkDefinition.addDeconvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Superseded by addDeconvolutionNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IDeconvolutionLayer |
VNetworkDefinition.addDeconvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDeconvolutionLayer |
VNetworkDefinition.addDeconvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDeconvolutionLayer |
INetworkDefinition.addDeconvolution(ITensor input,
int nbOutputMaps,
DimsHW kernelSize,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Superseded by addDeconvolutionNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IDeconvolutionLayer |
INetworkDefinition.addDeconvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDeconvolutionLayer |
VNetworkDefinition.addDeconvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDeconvolutionLayer |
VNetworkDefinition.addDeconvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDeconvolutionLayer |
INetworkDefinition.addDeconvolutionNd(ITensor input,
int nbOutputMaps,
Dims32 kernelSize,
Weights kernelWeights,
Weights biasWeights) |
IDequantizeLayer |
INetworkDefinition.addDequantize(ITensor input,
ITensor scale)
\brief Add a dequantization layer to the network.
|
IDequantizeLayer |
VNetworkDefinition.addDequantize(ITensor input,
ITensor scale) |
IDequantizeLayer |
VNetworkDefinition.addDequantize(ITensor input,
ITensor scale) |
IDequantizeLayer |
INetworkDefinition.addDequantize(ITensor input,
ITensor scale)
\brief Add a dequantization layer to the network.
|
IEinsumLayer |
INetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
BytePointer equation) |
IEinsumLayer |
VNetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
BytePointer equation) |
IEinsumLayer |
VNetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
BytePointer equation) |
IEinsumLayer |
INetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
BytePointer equation) |
IEinsumLayer |
INetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
String equation) |
IEinsumLayer |
VNetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
String equation) |
IEinsumLayer |
VNetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
String equation) |
IEinsumLayer |
INetworkDefinition.addEinsum(ITensor inputs,
int nbInputs,
String equation) |
IElementWiseLayer |
INetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
int op) |
IElementWiseLayer |
VNetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
int op) |
IElementWiseLayer |
VNetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
int op) |
IElementWiseLayer |
INetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
int op) |
IElementWiseLayer |
INetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
nvinfer.ElementWiseOperation op)
\brief Add an elementwise layer to the network.
|
IElementWiseLayer |
VNetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
nvinfer.ElementWiseOperation op) |
IElementWiseLayer |
VNetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
nvinfer.ElementWiseOperation op) |
IElementWiseLayer |
INetworkDefinition.addElementWise(ITensor input1,
ITensor input2,
nvinfer.ElementWiseOperation op)
\brief Add an elementwise layer to the network.
|
IFullyConnectedLayer |
INetworkDefinition.addFullyConnected(ITensor input,
int nbOutputs,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Use addMatrixMultiply instead. Deprecated in TensorRT 8.4.
|
IFullyConnectedLayer |
VNetworkDefinition.addFullyConnected(ITensor input,
int nbOutputs,
Weights kernelWeights,
Weights biasWeights) |
IFullyConnectedLayer |
VNetworkDefinition.addFullyConnected(ITensor input,
int nbOutputs,
Weights kernelWeights,
Weights biasWeights) |
IFullyConnectedLayer |
INetworkDefinition.addFullyConnected(ITensor input,
int nbOutputs,
Weights kernelWeights,
Weights biasWeights)
Deprecated.
Use addMatrixMultiply instead. Deprecated in TensorRT 8.4.
|
IGatherLayer |
INetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis)
\brief Add gather with mode GatherMode::kDEFAULT and specified axis and nbElementWiseDims=0.
|
IGatherLayer |
VNetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis) |
IGatherLayer |
VNetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis) |
IGatherLayer |
INetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis)
\brief Add gather with mode GatherMode::kDEFAULT and specified axis and nbElementWiseDims=0.
|
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode)
\brief Add gather with specified mode, axis=0 and nbElementWiseDims=0.
|
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode) |
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode) |
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode)
\brief Add gather with specified mode, axis=0 and nbElementWiseDims=0.
|
IIdentityLayer |
INetworkDefinition.addIdentity(ITensor input)
\brief Add an identity layer.
|
IIdentityLayer |
VNetworkDefinition.addIdentity(ITensor input) |
IIdentityLayer |
VNetworkDefinition.addIdentity(ITensor input) |
IIdentityLayer |
INetworkDefinition.addIdentity(ITensor input)
\brief Add an identity layer.
|
IIfConditionalInputLayer |
IIfConditional.addInput(ITensor input)
\brief Add an If-conditional input.
|
IIfConditionalInputLayer |
VIfConditional.addInput(ITensor tensor) |
IIfConditionalInputLayer |
VIfConditional.addInput(ITensor tensor) |
IIfConditionalInputLayer |
IIfConditional.addInput(ITensor input)
\brief Add an If-conditional input.
|
IIteratorLayer |
ILoop.addIterator(ITensor tensor) |
IIteratorLayer |
VLoop.addIterator(ITensor tensor) |
IIteratorLayer |
VLoop.addIterator(ITensor tensor) |
IIteratorLayer |
ILoop.addIterator(ITensor tensor) |
IIteratorLayer |
ILoop.addIterator(ITensor tensor,
int axis,
boolean reverse)
\brief Return layer that subscripts tensor by loop iteration.
|
IIteratorLayer |
VLoop.addIterator(ITensor tensor,
int axis,
boolean reverse) |
IIteratorLayer |
VLoop.addIterator(ITensor tensor,
int axis,
boolean reverse) |
IIteratorLayer |
ILoop.addIterator(ITensor tensor,
int axis,
boolean reverse)
\brief Return layer that subscripts tensor by loop iteration.
|
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
int outputKind) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
int outputKind) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
int outputKind) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
int outputKind) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
int outputKind,
int axis) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
int outputKind,
int axis) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
int outputKind,
int axis) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
int outputKind,
int axis) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind,
int axis)
\brief Make an output for this loop, based on the given tensor.
|
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind,
int axis) |
ILoopOutputLayer |
VLoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind,
int axis) |
ILoopOutputLayer |
ILoop.addLoopOutput(ITensor tensor,
nvinfer.LoopOutput outputKind,
int axis)
\brief Make an output for this loop, based on the given tensor.
|
ILRNLayer |
INetworkDefinition.addLRN(ITensor input,
int window,
float alpha,
float beta,
float k)
\brief Add a LRN layer to the network.
|
ILRNLayer |
VNetworkDefinition.addLRN(ITensor input,
int window,
float alpha,
float beta,
float k) |
ILRNLayer |
VNetworkDefinition.addLRN(ITensor input,
int window,
float alpha,
float beta,
float k) |
ILRNLayer |
INetworkDefinition.addLRN(ITensor input,
int window,
float alpha,
float beta,
float k)
\brief Add a LRN layer to the network.
|
IMatrixMultiplyLayer |
INetworkDefinition.addMatrixMultiply(ITensor input0,
int op0,
ITensor input1,
int op1) |
IMatrixMultiplyLayer |
VNetworkDefinition.addMatrixMultiply(ITensor input0,
int op0,
ITensor input1,
int op1) |
IMatrixMultiplyLayer |
VNetworkDefinition.addMatrixMultiply(ITensor input0,
int op0,
ITensor input1,
int op1) |
IMatrixMultiplyLayer |
INetworkDefinition.addMatrixMultiply(ITensor input0,
int op0,
ITensor input1,
int op1) |
IMatrixMultiplyLayer |
INetworkDefinition.addMatrixMultiply(ITensor input0,
nvinfer.MatrixOperation op0,
ITensor input1,
nvinfer.MatrixOperation op1)
\brief Add a MatrixMultiply layer to the network.
|
IMatrixMultiplyLayer |
VNetworkDefinition.addMatrixMultiply(ITensor input0,
nvinfer.MatrixOperation op0,
ITensor input1,
nvinfer.MatrixOperation op1) |
IMatrixMultiplyLayer |
VNetworkDefinition.addMatrixMultiply(ITensor input0,
nvinfer.MatrixOperation op0,
ITensor input1,
nvinfer.MatrixOperation op1) |
IMatrixMultiplyLayer |
INetworkDefinition.addMatrixMultiply(ITensor input0,
nvinfer.MatrixOperation op0,
ITensor input1,
nvinfer.MatrixOperation op1)
\brief Add a MatrixMultiply layer to the network.
|
IIfConditionalOutputLayer |
IIfConditional.addOutput(ITensor trueSubgraphOutput,
ITensor falseSubgraphOutput)
\brief Add an If-conditional output.
|
IIfConditionalOutputLayer |
VIfConditional.addOutput(ITensor trueTensor,
ITensor falseTensor) |
IIfConditionalOutputLayer |
VIfConditional.addOutput(ITensor trueTensor,
ITensor falseTensor) |
IIfConditionalOutputLayer |
IIfConditional.addOutput(ITensor trueSubgraphOutput,
ITensor falseSubgraphOutput)
\brief Add an If-conditional output.
|
IPaddingLayer |
INetworkDefinition.addPadding(ITensor input,
DimsHW prePadding,
DimsHW postPadding)
Deprecated.
Superseded by addPaddingNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IPaddingLayer |
VNetworkDefinition.addPadding(ITensor input,
DimsHW prePadding,
DimsHW postPadding) |
IPaddingLayer |
VNetworkDefinition.addPadding(ITensor input,
DimsHW prePadding,
DimsHW postPadding) |
IPaddingLayer |
INetworkDefinition.addPadding(ITensor input,
DimsHW prePadding,
DimsHW postPadding)
Deprecated.
Superseded by addPaddingNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IPaddingLayer |
INetworkDefinition.addPaddingNd(ITensor input,
Dims32 prePadding,
Dims32 postPadding)
Deprecated.
Superseded by addSlice. Deprecated in TensorRT 8.0
|
IPaddingLayer |
VNetworkDefinition.addPaddingNd(ITensor input,
Dims32 prePadding,
Dims32 postPadding) |
IPaddingLayer |
VNetworkDefinition.addPaddingNd(ITensor input,
Dims32 prePadding,
Dims32 postPadding) |
IPaddingLayer |
INetworkDefinition.addPaddingNd(ITensor input,
Dims32 prePadding,
Dims32 postPadding)
Deprecated.
Superseded by addSlice. Deprecated in TensorRT 8.0
|
IParametricReLULayer |
INetworkDefinition.addParametricReLU(ITensor input,
ITensor slope)
\brief Add a parametric ReLU layer to the network.
|
IParametricReLULayer |
VNetworkDefinition.addParametricReLU(ITensor input,
ITensor slope) |
IParametricReLULayer |
VNetworkDefinition.addParametricReLU(ITensor input,
ITensor slope) |
IParametricReLULayer |
INetworkDefinition.addParametricReLU(ITensor input,
ITensor slope)
\brief Add a parametric ReLU layer to the network.
|
IPluginV2Layer |
INetworkDefinition.addPluginV2(ITensor inputs,
int nbInputs,
IPluginV2 plugin) |
IPluginV2Layer |
VNetworkDefinition.addPluginV2(ITensor inputs,
int nbInputs,
IPluginV2 plugin) |
IPluginV2Layer |
VNetworkDefinition.addPluginV2(ITensor inputs,
int nbInputs,
IPluginV2 plugin) |
IPluginV2Layer |
INetworkDefinition.addPluginV2(ITensor inputs,
int nbInputs,
IPluginV2 plugin) |
IPoolingLayer |
INetworkDefinition.addPooling(ITensor input,
int type,
DimsHW windowSize)
Deprecated.
|
IPoolingLayer |
VNetworkDefinition.addPooling(ITensor input,
int type,
DimsHW windowSize) |
IPoolingLayer |
VNetworkDefinition.addPooling(ITensor input,
int type,
DimsHW windowSize) |
IPoolingLayer |
INetworkDefinition.addPooling(ITensor input,
int type,
DimsHW windowSize)
Deprecated.
|
IPoolingLayer |
INetworkDefinition.addPooling(ITensor input,
nvinfer.PoolingType type,
DimsHW windowSize)
Deprecated.
Superseded by addPoolingNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IPoolingLayer |
VNetworkDefinition.addPooling(ITensor input,
nvinfer.PoolingType type,
DimsHW windowSize) |
IPoolingLayer |
VNetworkDefinition.addPooling(ITensor input,
nvinfer.PoolingType type,
DimsHW windowSize) |
IPoolingLayer |
INetworkDefinition.addPooling(ITensor input,
nvinfer.PoolingType type,
DimsHW windowSize)
Deprecated.
Superseded by addPoolingNd. Deprecated prior to TensorRT 8.0 and will be removed in 9.0
|
IPoolingLayer |
INetworkDefinition.addPoolingNd(ITensor input,
int type,
Dims32 windowSize) |
IPoolingLayer |
VNetworkDefinition.addPoolingNd(ITensor input,
int type,
Dims32 windowSize) |
IPoolingLayer |
VNetworkDefinition.addPoolingNd(ITensor input,
int type,
Dims32 windowSize) |
IPoolingLayer |
INetworkDefinition.addPoolingNd(ITensor input,
int type,
Dims32 windowSize) |
IPoolingLayer |
INetworkDefinition.addPoolingNd(ITensor input,
nvinfer.PoolingType type,
Dims32 windowSize)
\brief Add a multi-dimension pooling layer to the network.
|
IPoolingLayer |
VNetworkDefinition.addPoolingNd(ITensor input,
nvinfer.PoolingType type,
Dims32 windowSize) |
IPoolingLayer |
VNetworkDefinition.addPoolingNd(ITensor input,
nvinfer.PoolingType type,
Dims32 windowSize) |
IPoolingLayer |
INetworkDefinition.addPoolingNd(ITensor input,
nvinfer.PoolingType type,
Dims32 windowSize)
\brief Add a multi-dimension pooling layer to the network.
|
IQuantizeLayer |
INetworkDefinition.addQuantize(ITensor input,
ITensor scale)
\brief Add a quantization layer to the network.
|
IQuantizeLayer |
VNetworkDefinition.addQuantize(ITensor input,
ITensor scale) |
IQuantizeLayer |
VNetworkDefinition.addQuantize(ITensor input,
ITensor scale) |
IQuantizeLayer |
INetworkDefinition.addQuantize(ITensor input,
ITensor scale)
\brief Add a quantization layer to the network.
|
IRaggedSoftMaxLayer |
INetworkDefinition.addRaggedSoftMax(ITensor input,
ITensor bounds)
\brief Add a RaggedSoftMax layer to the network.
|
IRaggedSoftMaxLayer |
VNetworkDefinition.addRaggedSoftMax(ITensor input,
ITensor bounds) |
IRaggedSoftMaxLayer |
VNetworkDefinition.addRaggedSoftMax(ITensor input,
ITensor bounds) |
IRaggedSoftMaxLayer |
INetworkDefinition.addRaggedSoftMax(ITensor input,
ITensor bounds)
\brief Add a RaggedSoftMax layer to the network.
|
IRecurrenceLayer |
ILoop.addRecurrence(ITensor initialValue)
\brief Create a recurrence layer for this loop with initialValue as its first input.
|
IRecurrenceLayer |
VLoop.addRecurrence(ITensor initialValue) |
IRecurrenceLayer |
VLoop.addRecurrence(ITensor initialValue) |
IRecurrenceLayer |
ILoop.addRecurrence(ITensor initialValue)
\brief Create a recurrence layer for this loop with initialValue as its first input.
|
IReduceLayer |
INetworkDefinition.addReduce(ITensor input,
int operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
VNetworkDefinition.addReduce(ITensor input,
int operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
VNetworkDefinition.addReduce(ITensor input,
int operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
INetworkDefinition.addReduce(ITensor input,
int operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
INetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions)
\brief Add a reduce layer to the network.
|
IReduceLayer |
VNetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
VNetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions) |
IReduceLayer |
INetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions)
\brief Add a reduce layer to the network.
|
IResizeLayer |
INetworkDefinition.addResize(ITensor input)
\brief Add a resize layer to the network.
|
IResizeLayer |
VNetworkDefinition.addResize(ITensor input) |
IResizeLayer |
VNetworkDefinition.addResize(ITensor input) |
IResizeLayer |
INetworkDefinition.addResize(ITensor input)
\brief Add a resize layer to the network.
|
IRNNv2Layer |
INetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
int op)
Deprecated.
|
IRNNv2Layer |
VNetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
int op) |
IRNNv2Layer |
VNetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
int op) |
IRNNv2Layer |
INetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
int op)
Deprecated.
|
IRNNv2Layer |
INetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
nvinfer.RNNOperation op)
Deprecated.
Superseded by INetworkDefinition::addLoop. Deprecated prior to TensorRT 8.0 and will be removed
in 9.0
\warning RNN inputs do not support wildcard dimensions or explicit batch size networks.
\warning Int32 tensors are not valid input tensors, only for sequence lengths.
|
IRNNv2Layer |
VNetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
nvinfer.RNNOperation op) |
IRNNv2Layer |
VNetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
nvinfer.RNNOperation op) |
IRNNv2Layer |
INetworkDefinition.addRNNv2(ITensor input,
int layerCount,
int hiddenSize,
int maxSeqLen,
nvinfer.RNNOperation op)
Deprecated.
Superseded by INetworkDefinition::addLoop. Deprecated prior to TensorRT 8.0 and will be removed
in 9.0
\warning RNN inputs do not support wildcard dimensions or explicit batch size networks.
\warning Int32 tensors are not valid input tensors, only for sequence lengths.
|
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power)
\brief Add a Scale layer to the network.
|
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power)
\brief Add a Scale layer to the network.
|
IScaleLayer |
INetworkDefinition.addScaleNd(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
INetworkDefinition.addScaleNd(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
INetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis)
\brief Add a multi-dimension scale layer to the network.
|
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
INetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis)
\brief Add a multi-dimension scale layer to the network.
|
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
int mode) |
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode)
\brief Add a Scatter layer to the network with specified mode and axis=0.
|
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode) |
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode) |
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode)
\brief Add a Scatter layer to the network with specified mode and axis=0.
|
ISelectLayer |
INetworkDefinition.addSelect(ITensor condition,
ITensor thenInput,
ITensor elseInput)
\brief Add a select layer to the network.
|
ISelectLayer |
VNetworkDefinition.addSelect(ITensor condition,
ITensor thenInput,
ITensor elseInput) |
ISelectLayer |
VNetworkDefinition.addSelect(ITensor condition,
ITensor thenInput,
ITensor elseInput) |
ISelectLayer |
INetworkDefinition.addSelect(ITensor condition,
ITensor thenInput,
ITensor elseInput)
\brief Add a select layer to the network.
|
IShapeLayer |
INetworkDefinition.addShape(ITensor input)
\brief Add a shape layer to the network.
|
IShapeLayer |
VNetworkDefinition.addShape(ITensor input) |
IShapeLayer |
VNetworkDefinition.addShape(ITensor input) |
IShapeLayer |
INetworkDefinition.addShape(ITensor input)
\brief Add a shape layer to the network.
|
IShuffleLayer |
INetworkDefinition.addShuffle(ITensor input)
\brief Add a shuffle layer to the network.
|
IShuffleLayer |
VNetworkDefinition.addShuffle(ITensor input) |
IShuffleLayer |
VNetworkDefinition.addShuffle(ITensor input) |
IShuffleLayer |
INetworkDefinition.addShuffle(ITensor input)
\brief Add a shuffle layer to the network.
|
ISliceLayer |
INetworkDefinition.addSlice(ITensor input,
Dims32 start,
Dims32 size,
Dims32 stride)
\brief Add a slice layer to the network.
|
ISliceLayer |
VNetworkDefinition.addSlice(ITensor input,
Dims32 start,
Dims32 size,
Dims32 stride) |
ISliceLayer |
VNetworkDefinition.addSlice(ITensor input,
Dims32 start,
Dims32 size,
Dims32 stride) |
ISliceLayer |
INetworkDefinition.addSlice(ITensor input,
Dims32 start,
Dims32 size,
Dims32 stride)
\brief Add a slice layer to the network.
|
ISoftMaxLayer |
INetworkDefinition.addSoftMax(ITensor input)
\brief Add a SoftMax layer to the network.
|
ISoftMaxLayer |
VNetworkDefinition.addSoftMax(ITensor input) |
ISoftMaxLayer |
VNetworkDefinition.addSoftMax(ITensor input) |
ISoftMaxLayer |
INetworkDefinition.addSoftMax(ITensor input)
\brief Add a SoftMax layer to the network.
|
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes)
\brief Add a TopK layer to the network.
|
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes) |
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes) |
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes)
\brief Add a TopK layer to the network.
|
ITripLimitLayer |
ILoop.addTripLimit(ITensor tensor,
int _limit) |
ITripLimitLayer |
VLoop.addTripLimit(ITensor tensor,
int _limit) |
ITripLimitLayer |
VLoop.addTripLimit(ITensor tensor,
int _limit) |
ITripLimitLayer |
ILoop.addTripLimit(ITensor tensor,
int _limit) |
ITripLimitLayer |
ILoop.addTripLimit(ITensor tensor,
nvinfer.TripLimit _limit)
\brief Add a trip-count limiter, based on the given tensor.
|
ITripLimitLayer |
VLoop.addTripLimit(ITensor tensor,
nvinfer.TripLimit _limit) |
ITripLimitLayer |
VLoop.addTripLimit(ITensor tensor,
nvinfer.TripLimit _limit) |
ITripLimitLayer |
ILoop.addTripLimit(ITensor tensor,
nvinfer.TripLimit _limit)
\brief Add a trip-count limiter, based on the given tensor.
|
IUnaryLayer |
INetworkDefinition.addUnary(ITensor input,
int operation) |
IUnaryLayer |
VNetworkDefinition.addUnary(ITensor input,
int operation) |
IUnaryLayer |
VNetworkDefinition.addUnary(ITensor input,
int operation) |
IUnaryLayer |
INetworkDefinition.addUnary(ITensor input,
int operation) |
IUnaryLayer |
INetworkDefinition.addUnary(ITensor input,
nvinfer.UnaryOperation operation)
\brief Add a unary layer to the network.
|
IUnaryLayer |
VNetworkDefinition.addUnary(ITensor input,
nvinfer.UnaryOperation operation) |
IUnaryLayer |
VNetworkDefinition.addUnary(ITensor input,
nvinfer.UnaryOperation operation) |
IUnaryLayer |
INetworkDefinition.addUnary(ITensor input,
nvinfer.UnaryOperation operation)
\brief Add a unary layer to the network.
|
void |
INetworkDefinition.markOutput(ITensor tensor)
\brief Mark a tensor as a network output.
|
void |
VNetworkDefinition.markOutput(ITensor tensor) |
void |
VNetworkDefinition.markOutput(ITensor tensor) |
void |
INetworkDefinition.markOutput(ITensor tensor)
\brief Mark a tensor as a network output.
|
boolean |
INetworkDefinition.markOutputForShapes(ITensor tensor)
\brief Enable tensor's value to be computed by IExecutionContext::getShapeBinding.
|
boolean |
VNetworkDefinition.markOutputForShapes(ITensor tensor) |
boolean |
VNetworkDefinition.markOutputForShapes(ITensor tensor) |
boolean |
INetworkDefinition.markOutputForShapes(ITensor tensor)
\brief Enable tensor's value to be computed by IExecutionContext::getShapeBinding.
|
void |
INetworkDefinition.removeTensor(ITensor tensor)
\brief remove a tensor from the network definition.
|
void |
VNetworkDefinition.removeTensor(ITensor tensor) |
void |
VNetworkDefinition.removeTensor(ITensor tensor) |
void |
INetworkDefinition.removeTensor(ITensor tensor)
\brief remove a tensor from the network definition.
|
void |
IRNNv2Layer.setCellState(ITensor cell)
Deprecated.
\brief Set the initial cell state of the LSTM with the provided \p cell ITensor.
|
void |
VRNNv2Layer.setCellState(ITensor cell) |
void |
VRNNv2Layer.setCellState(ITensor cell) |
void |
IRNNv2Layer.setCellState(ITensor cell)
Deprecated.
\brief Set the initial cell state of the LSTM with the provided \p cell ITensor.
|
IConditionLayer |
IIfConditional.setCondition(ITensor condition)
\brief Set the condition tensor for this If-Conditional construct.
|
IConditionLayer |
VIfConditional.setCondition(ITensor tensor) |
IConditionLayer |
VIfConditional.setCondition(ITensor tensor) |
IConditionLayer |
IIfConditional.setCondition(ITensor condition)
\brief Set the condition tensor for this If-Conditional construct.
|
void |
IRNNv2Layer.setHiddenState(ITensor hidden)
Deprecated.
\brief Set the initial hidden state of the RNN with the provided \p hidden ITensor.
|
void |
VRNNv2Layer.setHiddenState(ITensor hidden) |
void |
VRNNv2Layer.setHiddenState(ITensor hidden) |
void |
IRNNv2Layer.setHiddenState(ITensor hidden)
Deprecated.
\brief Set the initial hidden state of the RNN with the provided \p hidden ITensor.
|
void |
ILayer.setInput(int index,
ITensor tensor)
\brief Replace an input of this layer with a specific tensor.
|
void |
VLayer.setInput(int index,
ITensor tensor) |
void |
VLayer.setInput(int index,
ITensor tensor) |
void |
ILayer.setInput(int index,
ITensor tensor)
\brief Replace an input of this layer with a specific tensor.
|
void |
IRNNv2Layer.setSequenceLengths(ITensor seqLengths)
Deprecated.
\brief Specify individual sequence lengths in the batch with the ITensor pointed to by
\p seqLengths.
|
void |
VRNNv2Layer.setSequenceLengths(ITensor seqLengths) |
void |
VRNNv2Layer.setSequenceLengths(ITensor seqLengths) |
void |
IRNNv2Layer.setSequenceLengths(ITensor seqLengths)
Deprecated.
\brief Specify individual sequence lengths in the batch with the ITensor pointed to by
\p seqLengths.
|
void |
INetworkDefinition.unmarkOutput(ITensor tensor)
\brief unmark a tensor as a network output.
|
void |
VNetworkDefinition.unmarkOutput(ITensor tensor) |
void |
VNetworkDefinition.unmarkOutput(ITensor tensor) |
void |
INetworkDefinition.unmarkOutput(ITensor tensor)
\brief unmark a tensor as a network output.
|
boolean |
INetworkDefinition.unmarkOutputForShapes(ITensor tensor)
\brief Undo markOutputForShapes.
|
boolean |
VNetworkDefinition.unmarkOutputForShapes(ITensor tensor) |
boolean |
VNetworkDefinition.unmarkOutputForShapes(ITensor tensor) |
boolean |
INetworkDefinition.unmarkOutputForShapes(ITensor tensor)
\brief Undo markOutputForShapes.
|
| Modifier and Type | Method and Description |
|---|---|
ITensor |
IBlobNameToTensor.find(BytePointer name) |
ITensor |
IBlobNameToTensor.find(BytePointer name) |
ITensor |
IBlobNameToTensor.find(String name)
\brief Given a blob name, returns a pointer to a ITensor object.
|
ITensor |
IBlobNameToTensor.find(String name)
\brief Given a blob name, returns a pointer to a ITensor object.
|
Copyright © 2022. All rights reserved.