From 5dd88ab93e644c525a21ba820456e87cf3a6fb22 Mon Sep 17 00:00:00 2001 From: Kevin May Date: Thu, 19 Dec 2019 10:26:42 +0000 Subject: IVGCVSW-4303 Re-enable NCHW on Depthwise Convolution * Also re-enable NCHW on BatchToSpace, SpaceToBatch and Transpose Conv Signed-off-by: Kevin May Change-Id: I386eb3d852978e7cc611efad4f9f0c982140865a --- ConversionUtils.hpp | 10 ---------- 1 file changed, 10 deletions(-) (limited to 'ConversionUtils.hpp') diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 300c06f1..bbd2f07a 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -3215,11 +3215,6 @@ bool ConvertBatchToSpaceNd(const HalOperation& operation, if (Is12Operand(*output)) { batchToSpaceNdDesc.m_DataLayout = OptionalDataLayout(operation, 2, model, data); - - if (batchToSpaceNdDesc.m_DataLayout == armnn::DataLayout::NCHW) - { - return Fail("%s: Operation has invalid inputs NCHW is not supported", __func__); - } } // Setting crops to 0,0 0,0 as it is not supported in Android NN API batchToSpaceNdDesc.m_Crops = {{0, 0}, {0, 0}}; @@ -3322,11 +3317,6 @@ bool ConvertSpaceToBatchNd(const HalOperation& operation, const HalModel& model, if (Is12Operand(*output)) { descriptor.m_DataLayout = OptionalDataLayout(operation, 3, model, data); - - if (descriptor.m_DataLayout == armnn::DataLayout::NCHW) - { - return Fail("%s: Operation has invalid inputs NCHW is not supported", __func__); - } } bool isSupported = false; -- cgit v1.2.1