nvinfer1::ILayer Class Reference

Base class for all layer classes in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ILayer:

Public Member Functions
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

Base class for all layer classes in a network definition.

Member Function Documentation

virtual ITensor* nvinfer1::ILayer::getInput ( int  index ) const

pure virtual

Get the layer input corresponding to the given index.

Parameters

index

The index of the input tensor.

Returns

The input tensor, or nullptr if the index is out of range.

virtual const char* nvinfer1::ILayer::getName ( ) const

pure virtual

Return the name of a layer.

See Also

setName()

virtual ITensor* nvinfer1::ILayer::getOutput ( int  index ) const

pure virtual

Get the layer output corresponding to the given index.

Returns

The indexed output tensor, or nullptr if the index is out of range.

virtual LayerType nvinfer1::ILayer::getType ( ) const

pure virtual

Return the type of a layer.

See Also

LayerType

virtual void nvinfer1::ILayer::setName ( const char *  name )

pure virtual

Set the name of a layer.

This method copies the name string.

See Also

getName()

---

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

• NvInfer.h