|
TensorRT
|
This is the complete list of members for nvinfer1::INetworkDefinition, including all inherited members.
| addActivation(ITensor &input, ActivationType type)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addConcatenation(ITensor *const *inputs, int nbInputs)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addConstant(Dims dimensions, Weights weights)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addConvolution(ITensor &input, int nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addDeconvolution(ITensor &input, int nbOutputMaps, DimsHW kernelSize, Weights kernelWeights, Weights biasWeights)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addElementWise(ITensor &input1, ITensor &input2, ElementWiseOperation op)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addFullyConnected(ITensor &input, int nbOutputs, Weights kernelWeights, Weights biasWeights)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addGather(ITensor &data, ITensor &indices, int axis)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addInput(const char *name, DataType type, Dims dimensions)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addLRN(ITensor &input, int window, float alpha, float beta, float k)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addMatrixMultiply(ITensor &input0, bool transpose0, ITensor &input1, bool transpose1)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addPadding(ITensor &input, DimsHW prePadding, DimsHW postPadding)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addPlugin(ITensor *const *inputs, int nbInputs, IPlugin &plugin)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addPluginExt(ITensor *const *inputs, int nbInputs, IPluginExt &plugin)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addPooling(ITensor &input, PoolingType type, DimsHW windowSize)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addRaggedSoftMax(ITensor &input, ITensor &bounds)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addReduce(ITensor &input, ReduceOperation operation, uint32_t reduceAxes, bool keepDimensions)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addRNN(ITensor &inputs, int layerCount, std::size_t hiddenSize, int maxSeqLen, RNNOperation op, RNNInputMode mode, RNNDirection dir, Weights weights, Weights bias)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addRNNv2(ITensor &input, int32_t layerCount, int32_t hiddenSize, int32_t maxSeqLen, RNNOperation op)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addScale(ITensor &input, ScaleMode mode, Weights shift, Weights scale, Weights power)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addShuffle(ITensor &input)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addSoftMax(ITensor &input)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addTopK(ITensor &input, TopKOperation op, int k, uint32_t reduceAxes)=0 | nvinfer1::INetworkDefinition | pure virtual |
| addUnary(ITensor &input, UnaryOperation operation)=0 | nvinfer1::INetworkDefinition | pure virtual |
| destroy()=0 | nvinfer1::INetworkDefinition | pure virtual |
| getConvolutionOutputDimensionsFormula() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getDeconvolutionOutputDimensionsFormula() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getInput(int index) const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getLayer(int index) const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getNbInputs() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getNbLayers() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getNbOutputs() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getOutput(int index) const =0 | nvinfer1::INetworkDefinition | pure virtual |
| getPoolingOutputDimensionsFormula() const =0 | nvinfer1::INetworkDefinition | pure virtual |
| markOutput(ITensor &tensor)=0 | nvinfer1::INetworkDefinition | pure virtual |
| setConvolutionOutputDimensionsFormula(IOutputDimensionsFormula *formula)=0 | nvinfer1::INetworkDefinition | pure virtual |
| setDeconvolutionOutputDimensionsFormula(IOutputDimensionsFormula *formula)=0 | nvinfer1::INetworkDefinition | pure virtual |
| setPoolingOutputDimensionsFormula(IOutputDimensionsFormula *formula)=0 | nvinfer1::INetworkDefinition | pure virtual |
| ~INetworkDefinition() (defined in nvinfer1::INetworkDefinition) | nvinfer1::INetworkDefinition | inlineprotectedvirtual |