ie_icnn_network_stats.hpp

Go to the documentation of this file.

 // Copyright (C) 2018-2019 Intel Corporation


 // SPDX-License-Identifier: Apache-2.0


 //


 


 /**


  * @brief This is a header file for the ICNNNetworkStats class


  * @file ie_icnn_network_stats.hpp


  */


 #pragma once


 

 #include <string>


 #include <memory>


 #include <limits>


 #include <vector>


 #include <map>


 #include "details/ie_irelease.hpp"


 

 namespace InferenceEngine {

 

 class NetworkNodeStats;

 

 using NetworkNodeStatsPtr = std::shared_ptr<NetworkNodeStats>;

 using NetworkNodeStatsWeakPtr = std::weak_ptr<NetworkNodeStats>;

 using NetworkStatsMap = std::map<std::string, NetworkNodeStatsPtr>;


 /**


  * @class ICNNNetworkStats


  * @brief This is the interface to describe the NN topology scoring statistics


  */


 class ICNNNetworkStats : public details::IRelease {

 public:

     virtual void setNodesStats(const NetworkStatsMap& stats) = 0;

     virtual const NetworkStatsMap& getNodesStats() const = 0;

 

     virtual bool isEmpty() const = 0;

 };

 

 

 class NetworkNodeStats {

 public:

     NetworkNodeStats() { }

     explicit NetworkNodeStats(int statCount) {

         float mn = (std::numeric_limits<float>::max)();

         float mx = (std::numeric_limits<float>::min)();

 

         for (int i = 0; i < statCount; i++) {

             _minOutputs.push_back(mn);

             _maxOutputs.push_back(mx);

         }

     }

 

 public:

     std::vector<float> _minOutputs;

     std::vector<float> _maxOutputs;

 };

 

 

 }  // namespace InferenceEngine