summaryrefslogtreecommitdiff
path: root/inference-engine/src/inference_engine/cpp_interfaces/impl/ie_memory_state_internal.hpp
blob: c6beec18ad1fc1bf7d4683cf39a0a01441c2bc30 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
// Copyright (C) 2018 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//

#pragma once
#include <string>
#include <cpp_interfaces/interface/ie_imemory_state_internal.hpp>


namespace InferenceEngine {

/**
 * @brief minimal interface for memory state implementation
 */
class MemoryStateInternal : public IMemoryStateInternal {
    std::string name;
    Blob::Ptr state;

 public:
    explicit MemoryStateInternal(std::string name) : name(name) {
    }
    std::string GetName() const override {
        return name;
    }
    void SetState(Blob::Ptr newState) override {
        state = newState;
    }
    Blob::CPtr GetLastState() const override {
        return state;
    }
};


}  // namespace InferenceEngine