1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index ca9f6940a6..c3a2eaffcb 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -140,6 +140,12 @@ public:
     void VisitSubtractionLayer(const IConnectableLayer* layer,
                                const char* name = nullptr) override;
 
+    void VisitTransposeConvolution2dLayer(const IConnectableLayer* layer,
+                                          const TransposeConvolution2dDescriptor& descriptor,
+                                          const ConstTensor& weights,
+                                          const Optional<ConstTensor>& biases,
+                                          const char* name = nullptr) override;
+
     /// Extract the quantized network
     INetworkPtr RetrieveFinalNetwork() { return std::move(m_QuantizedNetwork); }