blob: 7179a10ccd0dbda5c0e2fbffa267304f2034f3bf (
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
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// See LICENSE file in the project root for full license information.
//
#include "Observable.hpp"
namespace armnn
{
void AddedLayerObservable::Update(Layer* graphLayer)
{
m_ObservedObjects.emplace_back(graphLayer);
}
void ErasedLayerNamesObservable::Update(Layer* graphLayer)
{
auto& relatedLayerNames = graphLayer->GetRelatedLayerNames();
// If the erased layer has no related layers we take the erased layer's name
// Otherwise we need to preserve the related layer names,
// since we want to preserve the original graph's information
if (relatedLayerNames.empty())
{
m_ObservedObjects.emplace_back(graphLayer->GetName());
}
else
{
for (auto& relatedLayerName : relatedLayerNames)
{
m_ObservedObjects.emplace_back(relatedLayerName);
}
}
}
}
|