nvinfer1::IConstantLayer Class Reference

Layer that represents a constant value. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IConstantLayer:

Public Member Functions
virtual void	setWeights (Weights weights)=0
	Set the weights for the layer. More...
virtual Weights	getWeights () const =0
	Get the weights for the layer. More...
virtual void	setDimensions (Dims dimensions)=0
	Set the dimensions for the layer. More...
virtual Dims	getDimensions () const =0
	Get the dimensions for the layer. More...
Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...
virtual void	setName (const char *name)=0
	Set the name of a layer. More...
virtual const char *	getName () const =0
	Return the name of a layer. More...
virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.
virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...
virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.
virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...
virtual void	setInput (int index, ITensor &tensor)=0
	replace an input of this layer with a specific tensor More...
virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...
virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...
virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...
virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...
virtual void	setOutputType (int index, DataType dataType)=0
	Set the output type of this layer. More...
virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...
virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...
virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

Detailed Description

Layer that represents a constant value.

Warning

Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

virtual Dims nvinfer1::IConstantLayer::getDimensions ( ) const

pure virtual

Get the dimensions for the layer.

Returns

the dimensions for the layer

See Also

getDimensions

virtual Weights nvinfer1::IConstantLayer::getWeights ( ) const

pure virtual

Get the weights for the layer.

See Also

setWeights

virtual void nvinfer1::IConstantLayer::setDimensions ( Dims  dimensions )

pure virtual

Set the dimensions for the layer.

Parameters

dimensions

The dimensions of the layer

See Also

setDimensions

virtual void nvinfer1::IConstantLayer::setWeights ( Weights  weights )

pure virtual

Set the weights for the layer.

If weights.type is DataType::kINT32, the output is a tensor of 32-bit indices. Otherwise the output is a tensor of real values, and the output type will be FP32, FP16, or quantized INT8 following TensorRT's normal precision rules.

See Also

getWeights()

---

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

• NvInfer.h