ie_cnn_net_reader.h

Go to the documentation of this file.

 // Copyright (C) 2018 Intel Corporation


 // SPDX-License-Identifier: Apache-2.0


 //


 


 /**


  * @brief This is a header file for the Network reader class (wrapper) used to build networks from a given IR


  * @file ie_cnn_net_reader.h


  */


 #pragma once


 

 #include <string>


 #include <vector>


 #include <memory>


 #include <map>


 #include "ie_blob.h"


 #include "ie_cnn_network.h"


 #include "ie_common.h"


 #include "ie_icnn_net_reader.h"


 #include "details/ie_exception_conversion.hpp"


 

 namespace InferenceEngine {


 /**


  * @brief This is a wrapper class used to build and parse a network from the given IR.


  * All the methods here can throw exceptions.


  */


 class CNNNetReader {

 public:


     /**


      * @brief A smart pointer to this class


      */


     using Ptr = std::shared_ptr<CNNNetReader>;

 


     /**


      * @brief A default constructor


      */


     CNNNetReader() : actual(shared_from_irelease(InferenceEngine::CreateCNNNetReader())) {}

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::ReadNetwork


      */


     void ReadNetwork(const std::string &filepath) {

         CALL_STATUS_FNC(ReadNetwork, filepath.c_str());

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::ReadNetwork(const void*, size_t, ResponseDesc*)


      */


     void ReadNetwork(const void *model, size_t size) {

         CALL_STATUS_FNC(ReadNetwork, model, size);

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::SetWeights


      */


     void SetWeights(const TBlob<uint8_t>::Ptr &weights) const {

         CALL_STATUS_FNC(SetWeights, weights);

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::ReadWeights


      */


     void ReadWeights(const std::string &filepath) const {

         CALL_STATUS_FNC(ReadWeights, filepath.c_str());

     }

 


     /**


     * @brief Gets a copy of built network object


     * @return A copy of the CNNNetwork object to be loaded


      */


     CNNNetwork getNetwork() {

         // network obj are to be updated upon this call


         if (network.get() == nullptr) {

             try {

                 network.reset(new CNNNetwork(actual));

             } catch (...) {

                 THROW_IE_EXCEPTION << "Could not allocate memory";

             }

         }

         return *network.get();

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::isParseSuccess


      */


     bool isParseSuccess() const {

         CALL_FNC_NO_ARGS(isParseSuccess);

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::getDescription


      */


     std::string getDescription() const {

         CALL_STATUS_FNC_NO_ARGS(getDescription);

         return resp.msg;

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::getName


      */


     std::string getName() const {

         char name[64];

         CALL_STATUS_FNC(getName, name, sizeof(name) / sizeof(*name));

         return name;

     }

 


     /**


      * @brief Wraps original method


      * ICNNNetReader::getVersion


      */


     int getVersion() const {

         CALL_FNC_NO_ARGS(getVersion);

     }

 

 private:

     std::shared_ptr<ICNNNetReader> actual;

     std::shared_ptr<CNNNetwork> network;

 };

 }  // namespace InferenceEngine