This class implements a builder for IE Layer. More...
#include <ie_layer_builder.hpp>
Public Types | |
using | Ptr = std::shared_ptr< Layer > |
A shared pointer to the Layer builder. | |
using | CPtr = std::shared_ptr< const Layer > |
A shared pointer to the constant Layer builder. | |
![]() | |
using | CPtr = std::shared_ptr< const ILayer > |
A shared pointer to the const ILayer object. | |
Public Member Functions | |
Layer (const std::string &type, const std::string &name="") | |
The constructor creates a Layer builder with layer type and layer name. More... | |
Layer (const ILayer::CPtr &layer) | |
The constructor creates a Layer builder from shared pointer to constant ILayer. More... | |
Layer (idx_t id, const Layer &layer) | |
The constructor creates a Layer builder with layer ID and layer builder. More... | |
bool | operator== (const Layer &rhs) const |
Compares the given Layer builder with the current one. More... | |
idx_t | getId () const noexcept override |
Returns layer ID. More... | |
const std::string & | getName () const noexcept override |
Returns a constant reference to layer name. More... | |
Layer & | setName (const std::string &name) |
Sets layer name. More... | |
const std::string & | getType () const noexcept override |
Returns a constant reference to layer type. More... | |
Layer & | setType (const std::string &type) |
Sets layer type. More... | |
const std::map< std::string, Parameter > & | getParameters () const noexcept override |
Returns map of parameters. More... | |
std::map< std::string, Parameter > & | getParameters () |
Returns map of parameters. More... | |
Layer & | setParameters (const std::map< std::string, Parameter > ¶ms) |
Sets parameters for layer. More... | |
const std::vector< Port > & | getInputPorts () const noexcept override |
Returns vector of input ports. More... | |
std::vector< Port > & | getInputPorts () |
Returns vector of input ports. More... | |
Layer & | setInputPorts (const std::vector< Port > &ports) |
Sets input ports. More... | |
const std::vector< Port > & | getOutputPorts () const noexcept override |
Returns vector of output ports. More... | |
std::vector< Port > & | getOutputPorts () |
Returns vector of output ports. More... | |
Layer & | setOutputPorts (const std::vector< Port > &ports) |
Sets output ports. More... | |
const ILayer::CPtr | build () const |
Validates the current builder and generates ILayer object. More... | |
void | validate (bool partial=false) const |
Validates layer builder. | |
![]() | |
virtual | ~ILayer ()=default |
Virtual destructor for the layer interface. | |
Static Public Member Functions | |
static void | addValidator (const std::string &type, const std::function< void(const Layer::CPtr &, bool)> &validator) |
Registers a new validator for type. More... | |
This class implements a builder for IE Layer.
|
explicit |
|
explicit |
|
static |
const ILayer::CPtr InferenceEngine::Builder::Layer::build | ( | ) | const |
|
overridevirtualnoexcept |
|
overridevirtualnoexcept |
std::vector<Port>& InferenceEngine::Builder::Layer::getInputPorts | ( | ) |
Returns vector of input ports.
|
overridevirtualnoexcept |
|
overridevirtualnoexcept |
std::vector<Port>& InferenceEngine::Builder::Layer::getOutputPorts | ( | ) |
Returns vector of output ports.
|
overridevirtualnoexcept |
std::map<std::string, Parameter>& InferenceEngine::Builder::Layer::getParameters | ( | ) |
Returns map of parameters.
|
overridevirtualnoexcept |
|
inline |
Layer& InferenceEngine::Builder::Layer::setName | ( | const std::string & | name | ) |
Layer& InferenceEngine::Builder::Layer::setParameters | ( | const std::map< std::string, Parameter > & | params | ) |
Sets parameters for layer.
params | constant map of parameters |
Layer& InferenceEngine::Builder::Layer::setType | ( | const std::string & | type | ) |