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 |