nvinfer1::apiv::VNetworkDefinition Class Referenceabstract
Inheritance diagram for nvinfer1::apiv::VNetworkDefinition:
Public Member Functions | |
| virtual ITensor * | addInput (const char *name, DataType type, Dims dimensions) noexcept=0 |
| virtual void | markOutput (ITensor &tensor) noexcept=0 |
| virtual IConvolutionLayer * | addConvolution (ITensor &input, int32_t nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0 |
| virtual IFullyConnectedLayer * | addFullyConnected (ITensor &input, int32_t nbOutputs, Weights kernelWeights, Weights biasWeights) noexcept=0 |
| virtual IActivationLayer * | addActivation (ITensor &input, ActivationType type) noexcept=0 |
| virtual IPoolingLayer * | addPooling (ITensor &input, PoolingType type, DimsHW windowSize) noexcept=0 |
| virtual ILRNLayer * | addLRN (ITensor &input, int32_t window, float alpha, float beta, float k) noexcept=0 |
| virtual IScaleLayer * | addScale (ITensor &input, ScaleMode mode, Weights shift, Weights scale, Weights power) noexcept=0 |
| virtual ISoftMaxLayer * | addSoftMax (ITensor &input) noexcept=0 |
| virtual IConcatenationLayer * | addConcatenation (ITensor *const *inputs, int32_t nbInputs) noexcept=0 |
| virtual IDeconvolutionLayer * | addDeconvolution (ITensor &input, int32_t nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0 |
| virtual IElementWiseLayer * | addElementWise (ITensor &input1, ITensor &input2, ElementWiseOperation op) noexcept=0 |
| virtual IUnaryLayer * | addUnary (ITensor &input, UnaryOperation operation) noexcept=0 |
| virtual IPaddingLayer * | addPadding (ITensor &input, DimsHW prePadding, DimsHW postPadding) noexcept=0 |
| virtual IShuffleLayer * | addShuffle (ITensor &input) noexcept=0 |
| virtual int32_t | getNbLayers () const noexcept=0 |
| virtual ILayer * | getLayer (int32_t index) const noexcept=0 |
| virtual int32_t | getNbInputs () const noexcept=0 |
| virtual ITensor * | getInput (int32_t index) const noexcept=0 |
| virtual int32_t | getNbOutputs () const noexcept=0 |
| virtual ITensor * | getOutput (int32_t index) const noexcept=0 |
| virtual IReduceLayer * | addReduce (ITensor &input, ReduceOperation operation, uint32_t reduceAxes, bool keepDimensions) noexcept=0 |
| virtual ITopKLayer * | addTopK (ITensor &input, TopKOperation op, int32_t k, uint32_t reduceAxes) noexcept=0 |
| virtual IGatherLayer * | addGather (ITensor &data, ITensor &indices, int32_t axis) noexcept=0 |
| virtual IRaggedSoftMaxLayer * | addRaggedSoftMax (ITensor &input, ITensor &bounds) noexcept=0 |
| virtual IMatrixMultiplyLayer * | addMatrixMultiply (ITensor &input0, MatrixOperation op0, ITensor &input1, MatrixOperation op1) noexcept=0 |
| virtual IConstantLayer * | addConstant (Dims dimensions, Weights weights) noexcept=0 |
| virtual IRNNv2Layer * | addRNNv2 (ITensor &input, int32_t layerCount, int32_t hiddenSize, int32_t maxSeqLen, RNNOperation op) noexcept=0 |
| virtual IIdentityLayer * | addIdentity (ITensor &input) noexcept=0 |
| virtual void | removeTensor (ITensor &tensor) noexcept=0 |
| virtual void | unmarkOutput (ITensor &tensor) noexcept=0 |
| virtual IPluginV2Layer * | addPluginV2 (ITensor *const *inputs, int32_t nbInputs, IPluginV2 &plugin) noexcept=0 |
| virtual ISliceLayer * | addSlice (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 IShapeLayer * | addShape (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 IParametricReLULayer * | addParametricReLU (ITensor &input, ITensor &slope) noexcept=0 |
| virtual IConvolutionLayer * | addConvolutionNd (ITensor &input, int32_t nbOutputMaps, Dims kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0 |
| virtual IPoolingLayer * | addPoolingNd (ITensor &input, PoolingType type, Dims windowSize) noexcept=0 |
| virtual IDeconvolutionLayer * | addDeconvolutionNd (ITensor &input, int32_t nbOutputMaps, Dims kernelSize, Weights kernelWeights, Weights biasWeights) noexcept=0 |
| virtual IScaleLayer * | addScaleNd (ITensor &input, ScaleMode mode, Weights shift, Weights scale, Weights power, int32_t channelAxis) noexcept=0 |
| virtual IResizeLayer * | addResize (ITensor &input) noexcept=0 |
| virtual bool | hasExplicitPrecision () const noexcept=0 |
| virtual ILoop * | addLoop () noexcept=0 |
| virtual ISelectLayer * | addSelect (ITensor &condition, ITensor &thenInput, ITensor &elseInput) noexcept=0 |
| virtual IFillLayer * | addFill (Dims dimensions, FillOperation op) noexcept=0 |
| virtual IPaddingLayer * | addPaddingNd (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 IErrorRecorder * | getErrorRecorder () const noexcept=0 |
| virtual IDequantizeLayer * | addDequantize (ITensor &input, ITensor &scale) noexcept=0 |
| virtual IQuantizeLayer * | addQuantize (ITensor &input, ITensor &scale) noexcept=0 |
The documentation for this class was generated from the following file:
