The class represents a builder for Eltwise layer. More...
#include <ie_eltwise_layer.hpp>
Public Types | |
enum | EltwiseType { SUM = 1, MAX, MUL, SUB, DIV, MIN, SQUARED_DIFF } |
The enum defines all Eltwise types. | |
Public Member Functions | |
EltwiseLayer (const std::string &name="") | |
The constructor creates a builder with the name. More... | |
EltwiseLayer (const Layer::Ptr &layer) | |
The constructor creates a builder from generic builder. More... | |
EltwiseLayer (const Layer::CPtr &layer) | |
The constructor creates a builder from generic builder. More... | |
EltwiseLayer & | setName (const std::string &name) |
Sets the name for the layer. More... | |
const std::vector< Port > & | getInputPorts () const |
Returns input ports. More... | |
EltwiseLayer & | setInputPorts (const std::vector< Port > &ports) |
Sets input ports. More... | |
const Port & | getOutputPort () const |
Returns output port. More... | |
EltwiseLayer & | setOutputPort (const Port &port) |
Sets output port. More... | |
EltwiseType | getEltwiseType () const |
Returns eltwise type. More... | |
EltwiseLayer & | setEltwiseType (EltwiseType type) |
Sets eltwise type. More... | |
const std::vector< float > | getScales () const |
Returns eltwise scales. More... | |
EltwiseLayer & | setScales (const std::vector< float > &scales) |
Sets eltwise scales. 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 Eltwise 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 |
EltwiseType InferenceEngine::Builder::EltwiseLayer::getEltwiseType | ( | ) | const |
Returns eltwise type.
const std::vector<Port>& InferenceEngine::Builder::EltwiseLayer::getInputPorts | ( | ) | const |
Returns input ports.
const Port& InferenceEngine::Builder::EltwiseLayer::getOutputPort | ( | ) | const |
Returns output port.
const std::vector<float> InferenceEngine::Builder::EltwiseLayer::getScales | ( | ) | const |
Returns eltwise scales.
EltwiseLayer& InferenceEngine::Builder::EltwiseLayer::setEltwiseType | ( | EltwiseType | type | ) |
Sets eltwise type.
type | Eltwise type |
EltwiseLayer& InferenceEngine::Builder::EltwiseLayer::setInputPorts | ( | const std::vector< Port > & | ports | ) |
Sets input ports.
ports | Vector of input ports |
EltwiseLayer& InferenceEngine::Builder::EltwiseLayer::setName | ( | const std::string & | name | ) |
EltwiseLayer& InferenceEngine::Builder::EltwiseLayer::setOutputPort | ( | const Port & | port | ) |
Sets output port.
port | Output port |
EltwiseLayer& InferenceEngine::Builder::EltwiseLayer::setScales | ( | const std::vector< float > & | scales | ) |
Sets eltwise scales.
scales | Vector of scales |