The class represents a builder for Pooling layer. More...
#include <ie_pooling_layer.hpp>


Public Types | |
| enum | PoolingType { MAX = 1, AVG = 2 } |
| The enum defines available pooling types. | |
| enum | RoundingType { CEIL = 1, FLOOR = 2 } |
| The enum defines available rounding types. | |
Public Member Functions | |
| PoolingLayer (const std::string &name="") | |
| The constructor creates a builder with the name. More... | |
| PoolingLayer (const Layer::Ptr &layer) | |
| The constructor creates a builder from generic builder. More... | |
| PoolingLayer (const Layer::CPtr &layer) | |
| The constructor creates a builder from generic builder. More... | |
| operator Layer () const override | |
| Operator creates generic layer builder. More... | |
| PoolingLayer & | setName (const std::string &name) |
| Sets the name for the layer. More... | |
| const Port & | getInputPort () const |
| Returns input port. More... | |
| PoolingLayer & | setInputPort (const Port &port) |
| Sets input port. More... | |
| const Port & | getOutputPort () const |
| Returns output port. More... | |
| PoolingLayer & | setOutputPort (const Port &port) |
| Sets output port. More... | |
| const std::vector< size_t > | getKernel () const |
| Returns kernel size. More... | |
| PoolingLayer & | setKernel (const std::vector< size_t > &kernel) |
| Sets kernel size. More... | |
| const std::vector< size_t > | getStrides () const |
| Returns vector of strides. More... | |
| PoolingLayer & | setStrides (const std::vector< size_t > &strides) |
| Sets strides. More... | |
| const std::vector< size_t > | getPaddingsBegin () const |
| Returns begin paddings. More... | |
| PoolingLayer & | setPaddingsBegin (const std::vector< size_t > &paddings) |
| Sets begin paddings. More... | |
| const std::vector< size_t > | getPaddingsEnd () const |
| Return end paddings. More... | |
| PoolingLayer & | setPaddingsEnd (const std::vector< size_t > &paddings) |
| Sets end paddings. More... | |
| PoolingType | getPoolingType () const |
| Returns pooling type. More... | |
| PoolingLayer & | setPoolingType (PoolingType type) |
| Sets pooling type. More... | |
| RoundingType | getRoundingType () const |
| Returns rounding type. More... | |
| PoolingLayer & | setRoundingType (RoundingType type) |
| Sets rounding types. More... | |
| bool | getExcludePad () const |
| Returns a type of pooling strategy. More... | |
| PoolingLayer & | setExcludePad (bool exclude) |
| Sets a type of pooling strategy. 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::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 Pooling 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 |
| bool InferenceEngine::Builder::PoolingLayer::getExcludePad | ( | ) | const |
Returns a type of pooling strategy.
| const Port& InferenceEngine::Builder::PoolingLayer::getInputPort | ( | ) | const |
Returns input port.
| const std::vector<size_t> InferenceEngine::Builder::PoolingLayer::getKernel | ( | ) | const |
Returns kernel size.
| const Port& InferenceEngine::Builder::PoolingLayer::getOutputPort | ( | ) | const |
Returns output port.
| const std::vector<size_t> InferenceEngine::Builder::PoolingLayer::getPaddingsBegin | ( | ) | const |
Returns begin paddings.
| const std::vector<size_t> InferenceEngine::Builder::PoolingLayer::getPaddingsEnd | ( | ) | const |
Return end paddings.
| PoolingType InferenceEngine::Builder::PoolingLayer::getPoolingType | ( | ) | const |
Returns pooling type.
| RoundingType InferenceEngine::Builder::PoolingLayer::getRoundingType | ( | ) | const |
Returns rounding type.
| const std::vector<size_t> InferenceEngine::Builder::PoolingLayer::getStrides | ( | ) | const |
Returns vector of strides.
|
overridevirtual |
Operator creates generic layer builder.
Reimplemented from InferenceEngine::Builder::LayerDecorator.
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setExcludePad | ( | bool | exclude | ) |
Sets a type of pooling strategy.
| exclude | zero-values in the padding are not used if true |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setInputPort | ( | const Port & | port | ) |
Sets input port.
| port | Input port |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setKernel | ( | const std::vector< size_t > & | kernel | ) |
Sets kernel size.
| kernel | Kernel size |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setName | ( | const std::string & | name | ) |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setOutputPort | ( | const Port & | port | ) |
Sets output port.
| port | Output port |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setPaddingsBegin | ( | const std::vector< size_t > & | paddings | ) |
Sets begin paddings.
| paddings | Vector of paddings |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setPaddingsEnd | ( | const std::vector< size_t > & | paddings | ) |
Sets end paddings.
| paddings | Vector of paddings |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setPoolingType | ( | PoolingType | type | ) |
Sets pooling type.
| type | Pooling type |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setRoundingType | ( | RoundingType | type | ) |
Sets rounding types.
| type | Rounding type |
| PoolingLayer& InferenceEngine::Builder::PoolingLayer::setStrides | ( | const std::vector< size_t > & | strides | ) |
Sets strides.
| strides | vector of strides |