commit bbf2e7477be984702e1a51f2a23910ee8349b867
author Adnan AlSinan <adnan.alsinan@arm.com> Wed Feb 22 12:15:14 2023 +0000
committer Adnan AlSinan <adnan.alsinan@arm.com> Wed Mar 01 11:48:22 2023 +0000
tree d5ce7a16e7f6e977bba2f88c8a913d071b4e04ed
parent 227db8db83cd85d8704a8edbd4e8c88af0dd1f84

Add support for kernel indices in Maxpool

- Add a max pooling implementation that returns kernel indices.
- Add a parameter in pooling info object to pick kernel indices impl.
- Add validation tests.

Resolves: [ONCPUML-1187]

Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: I485ef1604f676ee14d5f7f62d33699e49c38e4d3
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9192
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>

src/cpu/operators/CpuPool2d.h

diff --git a/src/cpu/operators/CpuPool2d.h b/src/cpu/operators/CpuPool2d.h
index 02c2609..5c571db 100644
--- a/src/cpu/operators/CpuPool2d.h
+++ b/src/cpu/operators/CpuPool2d.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021 Arm Limited.
+ * Copyright (c) 2021, 2023 Arm Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -76,6 +76,7 @@
     std::unique_ptr<INEKernel> _asm_glue;
 
     bool                             _is_global_pooling_layer;
+    bool                             _use_kernel_indices;
     DataLayout                       _data_layout;
     experimental::MemoryRequirements _aux_mem{};
 };