aboutsummaryrefslogtreecommitdiff
path: root/ConversionUtils.hpp
diff options
context:
space:
mode:
authorKevin May <kevin.may@arm.com>2019-12-19 10:26:42 +0000
committerKevin May <kevin.may@arm.com>2019-12-19 10:26:42 +0000
commit5dd88ab93e644c525a21ba820456e87cf3a6fb22 (patch)
tree21ba9ce60dffd4f9435e8803d7b72ca052ea5b94 /ConversionUtils.hpp
parentc05798ca177d6a198d02425f6c57a35ec9ed6468 (diff)
downloadandroid-nn-driver-5dd88ab93e644c525a21ba820456e87cf3a6fb22.tar.gz
IVGCVSW-4303 Re-enable NCHW on Depthwise Convolution
* Also re-enable NCHW on BatchToSpace, SpaceToBatch and Transpose Conv Signed-off-by: Kevin May <kevin.may@arm.com> Change-Id: I386eb3d852978e7cc611efad4f9f0c982140865a
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r--ConversionUtils.hpp10
1 files changed, 0 insertions, 10 deletions
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<HalPolicy>(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<HalPolicy>(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;