blob: 2216bbbf095158f778f8fa78160c7f92cc5b97cd (
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include "IGraphObservable.hpp"
#include "Graph.hpp"
namespace armnn
{
template <typename ObservedType>
class GraphObservable : public IGraphObservable
{
public:
using Iterator = typename std::list<ObservedType>::const_iterator;
GraphObservable(Graph& subject, GraphEvent notifyOnEvent)
: m_Subject(&subject)
{
m_NotifyOnEvent = notifyOnEvent;
m_Subject->AttachObservable(this, m_NotifyOnEvent);
};
void Clear() { m_ObservedObjects.clear(); };
Iterator begin() { return m_ObservedObjects.begin(); }
Iterator end() { return m_ObservedObjects.end(); }
protected:
~GraphObservable()
{
if (m_Subject)
{
m_Subject->DetachObservable(this, m_NotifyOnEvent);
}
}
GraphEvent m_NotifyOnEvent;
Graph* m_Subject;
std::list<ObservedType> m_ObservedObjects;
};
class AddedLayerObservable : public GraphObservable<Layer*>
{
public:
explicit AddedLayerObservable(Graph& subject)
: GraphObservable<Layer*>(subject, GraphEvent::LayerAdded)
{};
void Update(Layer* graphLayer) override;
};
class ErasedLayerNamesObservable : public GraphObservable<std::string>
{
public:
explicit ErasedLayerNamesObservable(Graph& subject)
: GraphObservable<std::string>(subject, GraphEvent::LayerErased)
{};
void Update(Layer* graphLayer) override;
};
} //namespace armnn
|