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 |
addIdentity(ITensor &input)=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 |
removeTensor(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 |
unmarkOutput(ITensor &tensor)=0 | nvinfer1::INetworkDefinition | pure virtual |
~INetworkDefinition() (defined in nvinfer1::INetworkDefinition) | nvinfer1::INetworkDefinition | inlineprotectedvirtual |