This is the main interface to describe the NN topology. More...
#include <ie_icnn_network.hpp>
Public Types | |
using | Ptr = std::shared_ptr< ICNNNetwork > |
A shared pointer to a ICNNNetwork interface. | |
using | InputShapes = std::map< std::string, SizeVector > |
Map of pairs: name of corresponding data and its dimension. | |
Public Member Functions | |
virtual std::shared_ptr< ngraph::Function > | getFunction () noexcept=0 |
Returns nGraph function. More... | |
virtual std::shared_ptr< const ngraph::Function > | getFunction () const noexcept=0 |
Returns constant nGraph function. More... | |
virtual void | getOutputsInfo (OutputsDataMap &out) const noexcept=0 |
Gets the network output Data node information. The received info is stored in the given Data node. More... | |
virtual void | getInputsInfo (InputsDataMap &inputs) const noexcept=0 |
Gets the network input Data node information. The received info is stored in the given InputsDataMap object. More... | |
virtual InputInfo::Ptr | getInput (const std::string &inputName) const noexcept=0 |
Returns information on certain input pointed by inputName. More... | |
virtual const std::string & | getName () const noexcept=0 |
Returns the network name. More... | |
virtual size_t | layerCount () const noexcept=0 |
Returns the number of layers in the network as an integer value. More... | |
virtual StatusCode | addOutput (const std::string &layerName, size_t outputIndex=0, ResponseDesc *resp=nullptr) noexcept=0 |
Adds output to the layer. More... | |
virtual StatusCode | setBatchSize (size_t size, ResponseDesc *responseDesc) noexcept=0 |
Changes the inference batch size. More... | |
virtual size_t | getBatchSize () const noexcept=0 |
Gets the inference batch size. More... | |
virtual StatusCode | reshape (const InputShapes &inputShapes, ResponseDesc *resp) noexcept |
Run shape inference with new input shapes for the network. More... | |
virtual StatusCode | serialize (const std::string &xmlPath, const std::string &binPath, ResponseDesc *resp) const noexcept=0 |
Serialize network to IR and weights files. More... | |
virtual | ~ICNNNetwork () |
A virtual destructor. | |
This is the main interface to describe the NN topology.
|
pure virtualnoexcept |
Adds output to the layer.
layerName | Name of the layer |
outputIndex | Index of the output |
resp | Response message |
|
pure virtualnoexcept |
Gets the inference batch size.
|
pure virtualnoexcept |
Returns constant nGraph function.
|
pure virtualnoexcept |
Returns nGraph function.
|
pure virtualnoexcept |
Returns information on certain input pointed by inputName.
inputName | Name of input layer to get info on |
|
pure virtualnoexcept |
Gets the network input Data node information. The received info is stored in the given InputsDataMap object.
For single and multiple inputs networks. This method need to be called to find out input names for using them later when calling InferenceEngine::InferRequest::SetBlob
inputs | Reference to InputsDataMap object. |
|
pure virtualnoexcept |
Returns the network name.
|
pure virtualnoexcept |
Gets the network output Data node information. The received info is stored in the given Data node.
For single and multiple outputs networks.
This method need to be called to find output names for using them later when calling InferenceEngine::InferRequest::GetBlob or InferenceEngine::InferRequest::SetBlob
out | Reference to the OutputsDataMap object |
|
pure virtualnoexcept |
Returns the number of layers in the network as an integer value.
|
inlinevirtualnoexcept |
Run shape inference with new input shapes for the network.
inputShapes | - map of pairs: name of corresponding data and its dimension. |
resp | Pointer to the response message that holds a description of an error if any occurred |
|
pure virtualnoexcept |
Serialize network to IR and weights files.
xmlPath | Path to output IR file. |
binPath | Path to output weights file. |
resp | Pointer to the response message that holds a description of an error if any occurred |
|
pure virtualnoexcept |
Changes the inference batch size.
size | Size of batch to set |
responseDesc | Pointer to the response message that holds a description of an error if any occurred |