diff options
Diffstat (limited to 'include/armnn/LayerVisitorBase.hpp')
-rw-r--r-- | include/armnn/LayerVisitorBase.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/include/armnn/LayerVisitorBase.hpp b/include/armnn/LayerVisitorBase.hpp index 48fc2bbb0b..8406efe068 100644 --- a/include/armnn/LayerVisitorBase.hpp +++ b/include/armnn/LayerVisitorBase.hpp @@ -193,6 +193,13 @@ public: void VisitSwitchLayer(const IConnectableLayer*, const char*) override { DefaultPolicy::Apply(__func__); } + + void VisitTransposeConvolution2dLayer(const IConnectableLayer*, + const TransposeConvolution2dDescriptor&, + const ConstTensor&, + const Optional<ConstTensor>&, + const char*) override { DefaultPolicy::Apply(__func__); } + }; -} //namespace armnn +} // namespace armnn |