This class contains all the information about the Neural Network and the related binary information. More...
#include <ie_cnn_network.h>
Public Member Functions | |
CNNNetwork ()=default | |
A default constructor. | |
CNNNetwork (std::shared_ptr< ICNNNetwork > network) | |
Allows helper class to manage lifetime of network object. More... | |
CNNNetwork (const std::shared_ptr< ngraph::Function > &network, const std::vector< IExtensionPtr > &exts={}) | |
A constructor from ngraph::Function object This constructor wraps existing ngraph::Function If you want to avoid modification of original Function, please create a copy. More... | |
virtual | ~CNNNetwork () |
A destructor. | |
virtual OutputsDataMap | getOutputsInfo () const |
Gets the network output Data node information. The received info is stored in the given Data node. More... | |
virtual InputsDataMap | getInputsInfo () const |
Gets the network input Data node information. The received info is stored in the given InputsDataMap object. More... | |
size_t | layerCount () const |
Returns the number of layers in the network as an integer value. More... | |
const std::string & | getName () const |
Returns the network name. More... | |
virtual void | setBatchSize (const size_t size) |
Changes the inference batch size. More... | |
virtual size_t | getBatchSize () const |
Gets the inference batch size. More... | |
operator ICNNNetwork::Ptr () | |
An overloaded operator cast to get pointer on current network. More... | |
operator ICNNNetwork & () | |
An overloaded operator & to get current network. More... | |
operator const ICNNNetwork & () const | |
An overloaded operator & to get current network. More... | |
std::shared_ptr< ngraph::Function > | getFunction () |
Returns constant nGraph function. More... | |
std::shared_ptr< const ngraph::Function > | getFunction () const |
Returns constant nGraph function. More... | |
void | addOutput (const std::string &layerName, size_t outputIndex=0) |
Adds output to the layer. More... | |
virtual ICNNNetwork::InputShapes | getInputShapes () const |
Helper method to get collect all input shapes with names of corresponding Data objects. More... | |
virtual void | reshape (const ICNNNetwork::InputShapes &inputShapes) |
Run shape inference with new input shapes for the network. More... | |
void | serialize (const std::string &xmlPath, const std::string &binPath="") const |
Serialize network to IR and weights files. More... | |
Protected Attributes | |
std::shared_ptr< ICNNNetwork > | network |
Network extra interface, might be nullptr. | |
ICNNNetwork * | actual = nullptr |
A pointer to the current network. | |
DataPtr | output |
A pointer to output data. | |
This class contains all the information about the Neural Network and the related binary information.
|
inlineexplicit |
Allows helper class to manage lifetime of network object.
network | Pointer to the network object |
|
explicit |
A constructor from ngraph::Function object This constructor wraps existing ngraph::Function If you want to avoid modification of original Function, please create a copy.
network | Pointer to the ngraph::Function object |
exts | Vector of pointers to IE extension objects |
|
inline |
Adds output to the layer.
Wraps ICNNNetwork::addOutput
layerName | Name of the layer |
outputIndex | Index of the output |
|
inlinevirtual |
Gets the inference batch size.
Wraps ICNNNetwork::getBatchSize
|
inline |
Returns constant nGraph function.
|
inline |
Returns constant nGraph function.
|
inlinevirtual |
Helper method to get collect all input shapes with names of corresponding Data objects.
|
inlinevirtual |
Gets the network input Data node information. The received info is stored in the given InputsDataMap object.
Wraps ICNNNetwork::getInputsInfo
|
inline |
|
inlinevirtual |
Gets the network output Data node information. The received info is stored in the given Data node.
Wraps ICNNNetwork::getOutputsInfo
|
inline |
Returns the number of layers in the network as an integer value.
Wraps ICNNNetwork::layerCount
|
inline |
An overloaded operator & to get current network.
|
inline |
An overloaded operator & to get current network.
|
inline |
An overloaded operator cast to get pointer on current network.
|
inlinevirtual |
Run shape inference with new input shapes for the network.
inputShapes | - map of pairs: name of corresponding data and its dimension. |
|
inline |
Serialize network to IR and weights files.
xmlPath | Path to output IR file. |
binPath | Path to output weights file. The parameter is skipped in case of executable graph info serialization. |
|
inlinevirtual |
Changes the inference batch size.
Wraps ICNNNetwork::setBatchSize
size | Size of batch to set |