diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 2de031e94a..c2efa211a6 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -180,6 +180,7 @@ struct Pooling2dDescriptor , m_StrideY(0) , m_OutputShapeRounding(OutputShapeRounding::Floor) , m_PaddingMethod(PaddingMethod::Exclude) + , m_DataLayout(DataLayout::NCHW) {}; PoolingAlgorithm m_PoolType; @@ -193,6 +194,7 @@ struct Pooling2dDescriptor uint32_t m_StrideY; OutputShapeRounding m_OutputShapeRounding; PaddingMethod m_PaddingMethod; + DataLayout m_DataLayout; }; struct FullyConnectedDescriptor |