1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index 89d1932a08..db0134d7a4 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -56,6 +56,10 @@ public:
                                   const BatchToSpaceNdDescriptor& batchToSpaceNdDescriptor,
                                   const char* name = nullptr) override;
 
+    void VisitComparisonLayer(const IConnectableLayer* layer,
+                              const ComparisonDescriptor& comparisonDescriptor,
+                              const char* name = nullptr) override;
+
     void VisitConcatLayer(const IConnectableLayer* layer,
                           const OriginsDescriptor& originsDescriptor,
                           const char* name = nullptr) override;