From 3883b2776cec33f16f0ea9a2d795de2b7c766df7 Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Fri, 22 Jul 2022 16:03:36 +0100 Subject: GitHub #667: Neon fold padding into average pool 2D quantization bug fix. * Originated from a GitHub issue: https://github.com/ARM-software/armnn/issues/667 * Initially, Arm NN supports the pool 2D operation because there is no padding on the pool2d. Neon failure occurs when padding is followed by average pool 2D due to folding optimization. * Here we prevent the folding optimization from happening for the above special case and add it in as a backend specific optimization. Signed-off-by: Cathal Corbett Change-Id: Ia0fd90c3a6b4b9d29c81106f154617d2e893e26b --- src/backends/reference/RefBackend.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/backends/reference/RefBackend.hpp') diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp index 9828d09f51..ecbe4d5ba9 100644 --- a/src/backends/reference/RefBackend.hpp +++ b/src/backends/reference/RefBackend.hpp @@ -50,7 +50,8 @@ public: IBackendInternal::ILayerSupportSharedPtr GetLayerSupport() const override; - OptimizationViews OptimizeSubgraphView(const SubgraphView& subgraph) const override; + OptimizationViews OptimizeSubgraphView(const SubgraphView& subgraph, + const ModelOptions& modelOptions) const override; std::vector GetHandleFactoryPreferences() const override; -- cgit v1.2.1