inference-engine/src/gna_plugin/gna_memory_state.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <memory>
   8 #include <cpp_interfaces/impl/ie_memory_state_internal.hpp>
   9 #include "gna_plugin.hpp"
  10
  11 namespace  GNAPluginNS {
  12
  13 class GNAMemoryState : public InferenceEngine::MemoryStateInternal {
  14     std::shared_ptr<GNAPlugin> plg;
  15  public:
  16     using Ptr = InferenceEngine::MemoryStateInternal::Ptr;
  17
  18     explicit GNAMemoryState(std::shared_ptr<GNAPlugin> plg)
  19         : InferenceEngine::MemoryStateInternal("GNAResetState"), plg(plg) {}
  20     void Reset() override {
  21         plg->Reset();
  22     }
  23 };
  24
  25 }  // namespace GNAPluginNS