diff options
author | morgolock <pablo.tello@arm.com> | 2020-04-09 14:17:48 +0100 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2020-06-15 14:04:49 +0000 |
commit | 37722d9a81627520fa347eb65199dbfeb84b26bd (patch) | |
tree | 3cb811c83e933337e685606625fcd44690b570d7 /src/runtime/NEON/functions/NEPoolingLayer.cpp | |
parent | 4a61653202afb018f4f259d3c144a735d73f0a20 (diff) | |
download | ComputeLibrary-37722d9a81627520fa347eb65199dbfeb84b26bd.tar.gz |
COMPMID-2449: Implement NEUnPoolLayer
Change-Id: I5677c87bba97dd395a3e13dbce34a3dd2c437033
Signed-off-by: morgolock <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3289
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEPoolingLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEPoolingLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEPoolingLayer.cpp b/src/runtime/NEON/functions/NEPoolingLayer.cpp index 12921cf40e..fb1b04ab6f 100644 --- a/src/runtime/NEON/functions/NEPoolingLayer.cpp +++ b/src/runtime/NEON/functions/NEPoolingLayer.cpp @@ -49,8 +49,8 @@ void NEPoolingLayer::configure(ITensor *input, ITensor *output, const PoolingLay case DataLayout::NCHW: { // Configure border depending on operation required (quantize border in case of asymmetric data_type) - BorderMode border_mode = (pool_info.pool_type == PoolingType::MAX) ? BorderMode::REPLICATE : BorderMode::CONSTANT; - PixelValue zero_value(0.f); + BorderMode border_mode = (!indices && pool_info.pool_type == PoolingType::MAX) ? BorderMode::REPLICATE : BorderMode::CONSTANT; + PixelValue zero_value((indices) ? std::numeric_limits<int>::min() : 0.f); if(is_data_type_quantized_asymmetric(input->info()->data_type()) && !pool_info.exclude_padding) { zero_value = PixelValue(0, input->info()->data_type(), input->info()->quantization_info()); |