The class represents a builder for ArgMax layer. More...
#include <ie_argmax_layer.hpp>


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 |
The class represents a builder for ArgMax layer.
|
explicit |
The constructor creates a builder with the name.
| name | Layer name |
|
explicit |
The constructor creates a builder from generic builder.
| layer | pointer to generic builder |
|
explicit |
The constructor creates a builder from generic builder.
| layer | constant pointer to generic builder |
| int InferenceEngine::Builder::ArgMaxLayer::getAxis | ( | ) | const |
Returns axis.
| size_t InferenceEngine::Builder::ArgMaxLayer::getOutMaxVal | ( | ) | const |
Returns output maximum value.
| const Port& InferenceEngine::Builder::ArgMaxLayer::getPort | ( | ) | const |
Returns port with shapes for the layer.
| size_t InferenceEngine::Builder::ArgMaxLayer::getTopK | ( | ) | const |
Returns top K.
| ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setAxis | ( | int | axis | ) |
Sets axis.
| axis | Axis |
| ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setName | ( | const std::string & | name | ) |
| ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setOutMaxVal | ( | size_t | size | ) |
Sets output maximum value.
| size | Maximum value |
| ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setPort | ( | const Port & | port | ) |
| ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setTopK | ( | size_t | topK | ) |
Sets top K.
| topK | Top K |