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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IElementWiseLayer:

Public Member Functions
virtual void	setOperation (ElementWiseOperation type)=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...

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.

Member Function Documentation

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

pure virtual

Get the binary operation for the layer.

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

pure virtual

Set the binary operation for the layer.

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