TensorRT  8.0.2
nvinfer1::apiv::VNetworkDefinition Class Referenceabstract
Inheritance diagram for nvinfer1::apiv::VNetworkDefinition:
nvinfer1::apiv::VRoot

Public Member Functions

virtual ITensoraddInput (const char *name, DataType type, Dims dimensions) noexcept=0
 
virtual void markOutput (ITensor &tensor) noexcept=0
 
virtual IConvolutionLayeraddConvolution (ITensor &input, int32_t nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0
 
virtual IFullyConnectedLayeraddFullyConnected (ITensor &input, int32_t nbOutputs, Weights kernelWeights, Weights biasWeights) noexcept=0
 
virtual IActivationLayeraddActivation (ITensor &input, ActivationType type) noexcept=0
 
virtual IPoolingLayeraddPooling (ITensor &input, PoolingType type, DimsHW windowSize) noexcept=0
 
virtual ILRNLayeraddLRN (ITensor &input, int32_t window, float alpha, float beta, float k) noexcept=0
 
virtual IScaleLayeraddScale (ITensor &input, ScaleMode mode, Weights shift, Weights scale, Weights power) noexcept=0
 
virtual ISoftMaxLayeraddSoftMax (ITensor &input) noexcept=0
 
virtual IConcatenationLayeraddConcatenation (ITensor *const *inputs, int32_t nbInputs) noexcept=0
 
virtual IDeconvolutionLayeraddDeconvolution (ITensor &input, int32_t nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0
 
virtual IElementWiseLayeraddElementWise (ITensor &input1, ITensor &input2, ElementWiseOperation op) noexcept=0
 
virtual IUnaryLayeraddUnary (ITensor &input, UnaryOperation operation) noexcept=0
 
virtual IPaddingLayeraddPadding (ITensor &input, DimsHW prePadding, DimsHW postPadding) noexcept=0
 
virtual IShuffleLayeraddShuffle (ITensor &input) noexcept=0
 
virtual int32_t getNbLayers () const noexcept=0
 
virtual ILayergetLayer (int32_t index) const noexcept=0
 
virtual int32_t getNbInputs () const noexcept=0
 
virtual ITensorgetInput (int32_t index) const noexcept=0
 
virtual int32_t getNbOutputs () const noexcept=0
 
virtual ITensorgetOutput (int32_t index) const noexcept=0
 
virtual IReduceLayeraddReduce (ITensor &input, ReduceOperation operation, uint32_t reduceAxes, bool keepDimensions) noexcept=0
 
virtual ITopKLayeraddTopK (ITensor &input, TopKOperation op, int32_t k, uint32_t reduceAxes) noexcept=0
 
virtual IGatherLayeraddGather (ITensor &data, ITensor &indices, int32_t axis) noexcept=0
 
virtual IRaggedSoftMaxLayeraddRaggedSoftMax (ITensor &input, ITensor &bounds) noexcept=0
 
virtual IMatrixMultiplyLayeraddMatrixMultiply (ITensor &input0, MatrixOperation op0, ITensor &input1, MatrixOperation op1) noexcept=0
 
virtual IConstantLayeraddConstant (Dims dimensions, Weights weights) noexcept=0
 
virtual IRNNv2LayeraddRNNv2 (ITensor &input, int32_t layerCount, int32_t hiddenSize, int32_t maxSeqLen, RNNOperation op) noexcept=0
 
virtual IIdentityLayeraddIdentity (ITensor &input) noexcept=0
 
virtual void removeTensor (ITensor &tensor) noexcept=0
 
virtual void unmarkOutput (ITensor &tensor) noexcept=0
 
virtual IPluginV2LayeraddPluginV2 (ITensor *const *inputs, int32_t nbInputs, IPluginV2 &plugin) noexcept=0
 
virtual ISliceLayeraddSlice (ITensor &input, Dims start, Dims size, Dims stride) noexcept=0
 
virtual void setName (const char *name) noexcept=0
 
virtual const char * getName () const noexcept=0
 
virtual IShapeLayeraddShape (ITensor &input) noexcept=0
 
virtual bool hasImplicitBatchDimension () const noexcept=0
 
virtual bool markOutputForShapes (ITensor &tensor) noexcept=0
 
virtual bool unmarkOutputForShapes (ITensor &tensor) noexcept=0
 
virtual IParametricReLULayeraddParametricReLU (ITensor &input, ITensor &slope) noexcept=0
 
virtual IConvolutionLayeraddConvolutionNd (ITensor &input, int32_t nbOutputMaps, Dims kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0
 
virtual IPoolingLayeraddPoolingNd (ITensor &input, PoolingType type, Dims windowSize) noexcept=0
 
virtual IDeconvolutionLayeraddDeconvolutionNd (ITensor &input, int32_t nbOutputMaps, Dims kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0
 
virtual IScaleLayeraddScaleNd (ITensor &input, ScaleMode mode, Weights shift, Weights scale, Weights power, int32_t channelAxis) noexcept=0
 
virtual IResizeLayeraddResize (ITensor &input) noexcept=0
 
virtual bool hasExplicitPrecision () const noexcept=0
 
virtual ILoopaddLoop () noexcept=0
 
virtual ISelectLayeraddSelect (ITensor &condition, ITensor &thenInput, ITensor &elseInput) noexcept=0
 
virtual IFillLayeraddFill (Dims dimensions, FillOperation op) noexcept=0
 
virtual IPaddingLayeraddPaddingNd (ITensor &input, Dims prePadding, Dims postPadding) noexcept=0
 
virtual bool setWeightsName (Weights weights, const char *name) noexcept=0
 
virtual void setErrorRecorder (IErrorRecorder *recorder) noexcept=0
 
virtual IErrorRecordergetErrorRecorder () const noexcept=0
 
virtual IDequantizeLayeraddDequantize (ITensor &input, ITensor &scale) noexcept=0
 
virtual IQuantizeLayeraddQuantize (ITensor &input, ITensor &scale) noexcept=0
 

The documentation for this class was generated from the following file: