nvinfer1::IUnaryLayer Class Reference

Layer that represents an unary operation. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IUnaryLayer:

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

Layer that represents an unary operation.

Member Function Documentation

virtual UnaryOperation nvinfer1::IUnaryLayer::getOperation ( ) const

pure virtual

Get the unary operation for the layer.

See Also

setOperation(), UnaryOperation

virtual void nvinfer1::IUnaryLayer::setOperation ( UnaryOperation  op )

pure virtual

Set the unary operation for the layer.

See Also

getOperation(), UnaryOperation

---

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

• NvInfer.h