aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/LayerVisitorBase.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/LayerVisitorBase.hpp')
-rw-r--r--include/armnn/LayerVisitorBase.hpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/include/armnn/LayerVisitorBase.hpp b/include/armnn/LayerVisitorBase.hpp
index 657051f2fa..47a8384924 100644
--- a/include/armnn/LayerVisitorBase.hpp
+++ b/include/armnn/LayerVisitorBase.hpp
@@ -50,7 +50,7 @@ public:
const char*) override { DefaultPolicy::Apply(__func__); }
void VisitConcatLayer(const IConnectableLayer*,
- const OriginsDescriptor&,
+ const ConcatDescriptor&,
const char*) override { DefaultPolicy::Apply(__func__); }
void VisitConstantLayer(const IConnectableLayer*,
@@ -122,7 +122,7 @@ public:
const char*) override { DefaultPolicy::Apply(__func__); }
void VisitMergerLayer(const IConnectableLayer*,
- const OriginsDescriptor&,
+ const MergerDescriptor&,
const char*) override { DefaultPolicy::Apply(__func__); }
void VisitMinimumLayer(const IConnectableLayer*,