diff options
Diffstat (limited to 'include/armnn/LayerVisitorBase.hpp')
-rw-r--r-- | include/armnn/LayerVisitorBase.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/include/armnn/LayerVisitorBase.hpp b/include/armnn/LayerVisitorBase.hpp index 363a09154d..0739b43736 100644 --- a/include/armnn/LayerVisitorBase.hpp +++ b/include/armnn/LayerVisitorBase.hpp @@ -39,6 +39,10 @@ public: void VisitAdditionLayer(const IConnectableLayer*, const char*) override { DefaultPolicy::Apply(__func__); } + void VisitArgMinMaxLayer(const IConnectableLayer*, + const ArgMinMaxDescriptor&, + const char*) override { DefaultPolicy::Apply(__func__); } + void VisitBatchNormalizationLayer(const IConnectableLayer*, const BatchNormalizationDescriptor&, const ConstTensor&, |