aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/QuantizerVisitor.hpp
diff options
context:
space:
mode:
authorFrancisMurtagh <francis.murtagh@arm.com>2019-02-12 10:11:43 +0000
committerFrancisMurtagh <francis.murtagh@arm.com>2019-02-12 10:11:43 +0000
commit424951560f2948b49506f178352e788cbe680fd8 (patch)
treeac2b4d815f954ce799e85920648aec05a3f8bbda /src/armnn/QuantizerVisitor.hpp
parent8c1701a2d9c1da0e1decb2afdc2093aa88810242 (diff)
downloadarmnn-424951560f2948b49506f178352e788cbe680fd8.tar.gz
IVGCVSW-2623 Support static quantization of DepthwiseConv2d
Change-Id: Iab0d5aed243aca921661e4d39770fe02b1330442 Signed-off-by: FrancisMurtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
-rw-r--r--src/armnn/QuantizerVisitor.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index cf151baf3c..c55ef6dd11 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -58,6 +58,12 @@ public:
const Optional<ConstTensor>& biases,
const char* name = nullptr) override;
+ void VisitDepthwiseConvolution2dLayer(const IConnectableLayer* layer,
+ const DepthwiseConvolution2dDescriptor& desc,
+ const ConstTensor& weights,
+ const Optional<ConstTensor>& biases,
+ const char* name = nullptr) override;
+
void VisitSoftmaxLayer(const IConnectableLayer* layer,
const SoftmaxDescriptor& softmaxDescriptor,
const char* name = nullptr) override;