nvinfer1::IElementWiseLayer Class Reference

A elementwise layer in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IElementWiseLayer:

Public Member Functions
virtual void	setOperation (ElementWiseOperation op)=0
	Set the binary operation for the layer. More...
virtual ElementWiseOperation	getOperation () const =0
	Get the binary operation 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

A elementwise layer in a network definition.

This layer applies a per-element binary operation between corresponding elements of two tensors.

The input dimensions of the two input tensors must be equal, and the output tensor is the same size as each input.

Warning

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

Member Function Documentation

virtual ElementWiseOperation nvinfer1::IElementWiseLayer::getOperation ( ) const

pure virtual

Get the binary operation for the layer.

See also

setOperation(), ElementWiseOperation

setBiasWeights()

virtual void nvinfer1::IElementWiseLayer::setOperation ( ElementWiseOperation  op )

pure virtual

Set the binary operation for the layer.

DLA supports only kSUM, kPROD, kMAX and kMIN.

See also

getOperation(), ElementWiseOperation

getBiasWeights()

---

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

• NvInfer.h