diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2018-10-05 14:08:48 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
| commit | 043d0d0e07e0a29648894cb9ba68993e540bb94d (patch) | |
| tree | 8f757c83d11ba0472f6b636f8ce8fa5b2f3e620e /include | |
| parent | 7c22c70418192cf5803ae60dd2da041b0a6aafe5 (diff) | |
| download | armnn-043d0d0e07e0a29648894cb9ba68993e540bb94d.tar.gz | |
IVGCVSW-1860 Support NHWC for Pooling2D
Change-Id: I9059a292e47867df82a5efbba5808fd264949ec9
Diffstat (limited to 'include')
| -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 |