layer that represents a unary operation More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IUnaryLayer:

Public Member Functions
virtual void	setOperation (UnaryOperation op)=0
	set the binary operation for the layer More...

virtual UnaryOperation	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

layer that represents a unary operation

Member Function Documentation

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

pure virtual

get the binary operation for the layer

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

pure virtual

set the binary operation for the layer

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