1 files changed, 10 insertions, 0 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index 18b97f6e..afaf1af7 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -1430,6 +1430,11 @@ bool ConvertPooling2d(const HalOperation& operation,
         if (Is12Operand(*output))
         {
             desc.m_DataLayout = OptionalDataLayout<HalPolicy>(operation, 10, model, data);
+
+            if (desc.m_DataLayout == armnn::DataLayout::NCHW)
+            {
+                return Fail("%s: Operation has invalid inputs NCHW is not supported", operationName);
+            }
         }
     }
     else
@@ -1449,6 +1454,11 @@ bool ConvertPooling2d(const HalOperation& operation,
         if (Is12Operand(*output))
         {
             desc.m_DataLayout = OptionalDataLayout<HalPolicy>(operation, 7, model, data);
+
+            if (desc.m_DataLayout == armnn::DataLayout::NCHW)
+            {
+                return Fail("%s: Operation has invalid inputs NCHW is not supported", operationName);
+            }
         }
 
         const armnnUtils::DataLayoutIndexed dataLayout(desc.m_DataLayout);