TensorRT  6.0.0.6
nvinfer1::IReduceLayer Class Referenceabstract

Layer that represents a reduction operator. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IReduceLayer:
nvinfer1::ILayer

Public Member Functions

virtual void setOperation (ReduceOperation op) TRTNOEXCEPT=0
 Set the reduce operation for the layer. More...
 
virtual ReduceOperation getOperation () const TRTNOEXCEPT=0
 Get the reduce operation for the layer. More...
 
virtual void setReduceAxes (uint32_t reduceAxes) TRTNOEXCEPT=0
 Set the axes over which to reduce. More...
 
virtual uint32_t getReduceAxes () const TRTNOEXCEPT=0
 Get the axes over which to reduce for the layer. More...
 
virtual void setKeepDimensions (bool keepDimensions) TRTNOEXCEPT=0
 Set the boolean that specifies whether or not to keep the reduced dimensions for the layer. More...
 
virtual bool getKeepDimensions () const TRTNOEXCEPT=0
 Get the boolean that specifies whether or not to keep the reduced dimensions for the layer. More...
 
- Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType getType () const TRTNOEXCEPT=0
 Return the type of a layer. More...
 
virtual void setName (const char *name) TRTNOEXCEPT=0
 Set the name of a layer. More...
 
virtual const char * getName () const TRTNOEXCEPT=0
 Return the name of a layer. More...
 
virtual int getNbInputs () const TRTNOEXCEPT=0
 Get the number of inputs of a layer.
 
virtual ITensorgetInput (int index) const TRTNOEXCEPT=0
 Get the layer input corresponding to the given index. More...
 
virtual int getNbOutputs () const TRTNOEXCEPT=0
 Get the number of outputs of a layer.
 
virtual ITensorgetOutput (int index) const TRTNOEXCEPT=0
 Get the layer output corresponding to the given index. More...
 
virtual void setInput (int index, ITensor &tensor) TRTNOEXCEPT=0
 replace an input of this layer with a specific tensor More...
 
virtual void setPrecision (DataType dataType) TRTNOEXCEPT=0
 Set the computational precision of this layer. More...
 
virtual DataType getPrecision () const TRTNOEXCEPT=0
 get the computational precision of this layer More...
 
virtual bool precisionIsSet () const TRTNOEXCEPT=0
 whether the computational precision has been set for this layer More...
 
virtual void resetPrecision () TRTNOEXCEPT=0
 reset the computational precision for this layer More...
 
virtual void setOutputType (int index, DataType dataType) TRTNOEXCEPT=0
 Set the output type of this layer. More...
 
virtual DataType getOutputType (int index) const TRTNOEXCEPT=0
 get the output type of this layer More...
 
virtual bool outputTypeIsSet (int index) const TRTNOEXCEPT=0
 whether the output type has been set for this layer More...
 
virtual void resetOutputType (int index) TRTNOEXCEPT=0
 reset the output type for this layer More...
 

Detailed Description

Layer that represents a reduction operator.

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

Member Function Documentation

virtual bool nvinfer1::IReduceLayer::getKeepDimensions ( ) const
pure virtual

Get the boolean that specifies whether or not to keep the reduced dimensions for the layer.

See also
setKeepDimensions
virtual ReduceOperation nvinfer1::IReduceLayer::getOperation ( ) const
pure virtual

Get the reduce operation for the layer.

See also
setOperation(), ReduceOperation
virtual uint32_t nvinfer1::IReduceLayer::getReduceAxes ( ) const
pure virtual

Get the axes over which to reduce for the layer.

See also
setReduceAxes
virtual void nvinfer1::IReduceLayer::setKeepDimensions ( bool  keepDimensions)
pure virtual

Set the boolean that specifies whether or not to keep the reduced dimensions for the layer.

See also
getKeepDimensions
virtual void nvinfer1::IReduceLayer::setOperation ( ReduceOperation  op)
pure virtual

Set the reduce operation for the layer.

See also
getOperation(), ReduceOperation
virtual void nvinfer1::IReduceLayer::setReduceAxes ( uint32_t  reduceAxes)
pure virtual

Set the axes over which to reduce.

See also
getReduceAxes

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