InferenceEngine::Builder::ArgMaxLayer Class Reference

The class represents a builder for ArgMax layer. More...

#include <ie_argmax_layer.hpp>

Inheritance diagram for InferenceEngine::Builder::ArgMaxLayer:

Collaboration diagram for InferenceEngine::Builder::ArgMaxLayer:

Public Member Functions
	ArgMaxLayer (const std::string &name="")
	The constructor creates a builder with the name. More...
	ArgMaxLayer (const Layer::Ptr &layer)
	The constructor creates a builder from generic builder. More...
	ArgMaxLayer (const Layer::CPtr &layer)
	The constructor creates a builder from generic builder. More...
ArgMaxLayer &	setName (const std::string &name)
	Sets the name for the layer. More...
const Port &	getPort () const
	Returns port with shapes for the layer. More...
ArgMaxLayer &	setPort (const Port &port)
	Sets port shapes for the layer. More...
int	getAxis () const
	Returns axis. More...
ArgMaxLayer &	setAxis (int axis)
	Sets axis. More...
size_t	getTopK () const
	Returns top K. More...
ArgMaxLayer &	setTopK (size_t topK)
	Sets top K. More...
size_t	getOutMaxVal () const
	Returns output maximum value. More...
ArgMaxLayer &	setOutMaxVal (size_t size)
	Sets output maximum value. More...
Public Member Functions inherited from InferenceEngine::Builder::LayerDecorator
	LayerDecorator (const std::string &type, const std::string &name)
	The constructor creates layer builders with layer type and layer name. More...
	LayerDecorator (const Layer::Ptr &layer)
	The constructor creates layer builders from reference to generic layer builder. More...
	LayerDecorator (const Layer::CPtr &layer)
	The constructor creates layer builders from reference to generic layer builder. More...
	LayerDecorator (const LayerDecorator &rval)
	The copy constructor. More...
LayerDecorator &	operator= (const LayerDecorator &rval)
	Copy operator for LayerDecorator. More...
virtual	~LayerDecorator ()=default
	Virtual destructor.
virtual	operator Layer () const
	The operator creates generic builder. More...
virtual	operator Layer::Ptr ()
	The operator creates generic builder. More...
virtual	operator Layer::CPtr () const
	The operator creates generic builder. More...
const std::string &	getType () const
	Returns layer type. More...
const std::string &	getName () const
	Returns layer name. More...

Additional Inherited Members
Protected Member Functions inherited from InferenceEngine::Builder::LayerDecorator
Layer::Ptr &	getLayer ()
const Layer::CPtr	getLayer () const
void	checkType (const std::string &type) const
Protected Attributes inherited from InferenceEngine::Builder::LayerDecorator
Layer::CPtr	cLayer

Detailed Description

The class represents a builder for ArgMax layer.

Constructor & Destructor Documentation

ArgMaxLayer() [1/3]

InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer ( const std::string &  name = "" )

explicit

The constructor creates a builder with the name.

Parameters

name	Layer name

ArgMaxLayer() [2/3]

InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer ( const Layer::Ptr &  layer )

explicit

The constructor creates a builder from generic builder.

Parameters

layer

pointer to generic builder

ArgMaxLayer() [3/3]

InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer ( const Layer::CPtr &  layer )

explicit

The constructor creates a builder from generic builder.

Parameters

layer

constant pointer to generic builder

Member Function Documentation

getAxis()

int InferenceEngine::Builder::ArgMaxLayer::getAxis

(

)

const

Returns axis.

Returns

Axis

getOutMaxVal()

size_t InferenceEngine::Builder::ArgMaxLayer::getOutMaxVal

(

)

const

Returns output maximum value.

Returns

Output maximum value

getPort()

const Port& InferenceEngine::Builder::ArgMaxLayer::getPort

(

)

const

Returns port with shapes for the layer.

Returns

Port with shapes

getTopK()

size_t InferenceEngine::Builder::ArgMaxLayer::getTopK

(

)

const

Returns top K.

Returns

Top K

setAxis()

ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setAxis

(

int

axis

)

Sets axis.

Parameters

axis

Axis

Returns

reference to layer builder

setName()

ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setName

(

const std::string &

name

)

Sets the name for the layer.

Parameters

name	Layer name

Returns

reference to layer builder

setOutMaxVal()

ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setOutMaxVal

(

size_t

size

)

Sets output maximum value.

Parameters

size	Maximum value

Returns

reference to layer builder

setPort()

ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setPort

(

const Port &

port

)

Sets port shapes for the layer.

Parameters

port	Port with shapes

Returns

reference to layer builder

setTopK()

ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setTopK

(

size_t

topK

)

Sets top K.

Parameters

topK

Top K

Returns

reference to layer builder

---

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

• builders/ie_argmax_layer.hpp