diff options
| author | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-01-26 12:32:45 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:46:07 +0000 |
| commit | 6e464c37b5335e362ac3f988cc4b0beed5205ff4 (patch) | |
| tree | 24f758be47f1dd439a4ab3f8f3631a1ed7ef4566 /src/runtime/NEON/functions/NEPoolingLayer.cpp | |
| parent | 3364c4ca658d44f449a8d3d6e9eee31d90254f15 (diff) | |
| download | ComputeLibrary-6e464c37b5335e362ac3f988cc4b0beed5205ff4.tar.gz | |
COMPMID-828 - Add support for non square pool size - Part1
Change-Id: Ib8100e7c659c49694c746fa3f36ce20f44f6929f
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/117804
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEPoolingLayer.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEPoolingLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEPoolingLayer.cpp b/src/runtime/NEON/functions/NEPoolingLayer.cpp index 8a32507a73..bc0b6f86d3 100644 --- a/src/runtime/NEON/functions/NEPoolingLayer.cpp +++ b/src/runtime/NEON/functions/NEPoolingLayer.cpp @@ -38,7 +38,7 @@ NEPoolingLayer::NEPoolingLayer() void NEPoolingLayer::configure(ITensor *input, ITensor *output, const PoolingLayerInfo &pool_info) { // Check if we have Global Pooling Layer - _is_global_pooling_layer = (input->info()->dimension(0) == pool_info.pool_size()) && (input->info()->dimension(1) == pool_info.pool_size()); + _is_global_pooling_layer = (input->info()->dimension(0) == pool_info.pool_size().width) && (input->info()->dimension(1) == pool_info.pool_size().height); // Configure pooling kernel _pooling_layer_kernel.configure(input, output, pool_info); |