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, 4 insertions, 0 deletions
diff --git a/include/armnn/LayerVisitorBase.hpp b/include/armnn/LayerVisitorBase.hpp
index d626c712ba..65d23039b7 100644
--- a/include/armnn/LayerVisitorBase.hpp
+++ b/include/armnn/LayerVisitorBase.hpp
@@ -69,6 +69,10 @@ public:
const Optional<ConstTensor>&,
const char*) override { DefaultPolicy::Apply(__func__); }
+ void VisitDepthToSpaceLayer(const IConnectableLayer*,
+ const DepthToSpaceDescriptor&,
+ const char*) override { DefaultPolicy::Apply(__func__); }
+
void VisitDepthwiseConvolution2dLayer(const IConnectableLayer*,
const DepthwiseConvolution2dDescriptor&,
const ConstTensor&,