-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include <builders/ie_layer_builder.hpp>
#include <ie_icnn_network.hpp>
#include <cpp/ie_cnn_network.h>
-#include <ie_inetwork.hpp>
+#include <ie_network.hpp>
#include <ie_context.hpp>
#include <ie_common.h>
#include <ie_blob.h>
/**
* @brief This class implements a builder for IE Network
*/
-class INFERENCE_ENGINE_API_CLASS(Network) {
+class INFERENCE_ENGINE_API_CLASS(Network): public INetwork {
public:
/**
* @brief A shared pointer to the Network builder
*/
using Ptr = std::shared_ptr<Network>;
+ /**
+ * @brief An iterator for Network builder definition
+ */
+ using iterator = details::INetworkIterator<Network, Layer>;
+ /**
+ * @brief Begin network iterator
+ * @return Network iterator
+ */
+ iterator begin();
+ /**
+ * @brief Begin network iterator
+ * @return const INetwork iterator
+ */
+ const_iterator begin() const noexcept override;
+
+ /**
+ * @brief End network iterator
+ * @return Network iterator
+ */
+ iterator end();
+ /**
+ * @brief End network iterator
+ * @return const INetwork iterator
+ */
+ const_iterator end() const noexcept override;
+
+ /**
+ * @brief Returns a number of layers in the network.
+ * @return Layers count
+ */
+ size_t size() const noexcept override;
/**
* @brief The constructor creates a builder based on ICNNNetwork
Network(const Context& ieContext, const INetwork& network);
/**
- * @brief Virtual destructor
- */
- virtual ~Network() = default;
-
- /**
* @brief Adds new layer and connects it with previous layers
*
* @param inputs Vector with PortInfo objects from previous layers
void disconnect(const Connection& connection);
/**
- * @brief Returns layer builder by ID
- *
- * @param layerId Layer ID
+ * @brief Returns vector of layer builders
*
- * @return Layer buider
+ * @return Vector of layer builders
*/
- Layer& getLayer(idx_t layerId);
+ std::vector<Layer::Ptr>& getLayers();
/**
- * @brief Returns constant layer builder by ID
- *
- * @param layerId Layer ID
+ * @brief Returns constant vector of layer builders
*
- * @return constant layer builder
+ * @return constant vector of layer builders
*/
- const Layer& getLayer(idx_t layerId) const;
+ const std::vector<Layer::Ptr>& getLayers() const;
/**
- * @brief Returns vector of layer builders
- *
- * @return Vector of layer builders
+ * @brief Returns a constant smart pointer to a Layer interface.
+ * If the layer is missing, returns nullptr.
+ * @param id Id of the Layer
+ * @return Layer interface smart pointer
*/
- std::vector<Layer>& getLayers();
+ const ILayer::CPtr getLayer(idx_t id) const noexcept override;
+ Layer::Ptr getLayer(idx_t layerId);
+
/**
- * @brief Returns constant vector of layer builders
- *
- * @return constant vector of layer builders
+ * @brief Returns a constant vector of input layers.
+ * @return Vector of input layers
*/
- const std::vector<Layer>& getLayers() const;
+ const std::vector<ILayer::CPtr> getInputs() const noexcept override;
+ /**
+ * @brief Returns a vector of input layers.
+ * @return Vector of input layers
+ */
+ std::vector<Layer::Ptr> getInputs();
/**
- * @brief Returns all connections for layer
- *
- * @param layerId Layer ID
- *
- * @return Vector of connections for the current layer
+ * @brief Returns a constant vector of output layers.
+ * @return Vector of output layers
+ */
+ const std::vector<ILayer::CPtr> getOutputs() const noexcept override;
+ /**
+ * @brief Returns a vector of input layers.
+ * @return Vector of input layers
+ */
+ std::vector<Layer::Ptr> getOutputs();
+
+ /**
+ * @brief Returns a constant vector of connections for specific layer.
+ * If the layer is missing, returns empty vector.
+ * @param layerId layer index
+ * @return Vector of connections
*/
- const std::vector<Connection> getLayerConnections(idx_t layerId) const noexcept;
+ const std::vector<Connection> getLayerConnections(idx_t layerId) const noexcept override;
/**
- * @brief Builds and validate networks
+ * @brief Returns a constant vector of all connections.
+ * @return Vector of connections
+ */
+ const std::vector<Connection>& getConnections() const;
+
+ /**
+ * @brief Returns a network name.
+ * @return Network name
+ */
+ const std::string& getName() const noexcept override;
+
+ /**
+ * @brief Returns a network context
+ * @return const reference to Context
+ */
+ const Context& getContext() const noexcept override;
+ /**
+ * @brief Returns a network context
+ * @return reference to Context
+ */
+ Context& getContext() noexcept;
+
+ /**
+ * @brief Builds and validate network
*
* @return const shared pointer to INetwork
*/
- const INetwork::Ptr build() const;
+ const INetwork::CPtr build();
+
+ /**
+ * @brief Validates network
+ *
+ */
+ void validate();
/**
* @brief The operator builds network
*
* @return const shared pointer to INetwork
*/
- explicit operator const INetwork::Ptr() const;
+ explicit operator const INetwork::CPtr();
private:
- const Context ctx;
- const size_t version;
- std::string name;
- std::vector<Layer> layers;
- std::vector<Connection> connections;
+ std::map<std::string, Parameter> parameters;
};
/**
* @param network constant shared pointer to INetwork object
* @return constant shared pointer to ICNNNetwork
*/
-INFERENCE_ENGINE_API_CPP(const std::shared_ptr<ICNNNetwork>) convertToICNNNetwork(const INetwork::Ptr& network);
+INFERENCE_ENGINE_API_CPP(const std::shared_ptr<ICNNNetwork>) convertToICNNNetwork(const INetwork::CPtr& network);
} // namespace Builder
projects / platform / upstream / dldt.git / blobdiff