inference-engine/src/inference_engine/cnn_network_stats_impl.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <map>
   8 #include <memory>
   9 #include <ie_icnn_network.hpp>
  10 #include "ie_common.h"
  11 #include "ie_data.h"
  12 #include "ie_blob.h"
  13 #include "ie_api.h"
  14 #include "description_buffer.hpp"
  15 #include <string>
  16 #include <vector>
  17
  18 #include <ie_icnn_network_stats.hpp>
  19
  20 namespace InferenceEngine {
  21 namespace details {
  22 class INFERENCE_ENGINE_API_CLASS(CNNNetworkStatsImpl) : public ICNNNetworkStats {
  23 public:
  24     CNNNetworkStatsImpl() = default;
  25     virtual ~CNNNetworkStatsImpl();
  26 public:
  27     const NetworkStatsMap& getNodesStats() const override;
  28     void setNodesStats(const NetworkStatsMap& stats)override;
  29     bool isEmpty() const override { return netNodesStats.empty(); }
  30
  31     void Release() noexcept override {
  32         delete this;
  33     }
  34 protected:
  35     std::map<std::string, NetworkNodeStatsPtr> netNodesStats;
  36 };
  37
  38 typedef std::shared_ptr<CNNNetworkStatsImpl> CNNNetworkStatsImplPtr;
  39 }  // namespace details
  40 }  // namespace InferenceEngine