commit 7930db48e12dd3a14c1971f41f5b83527efea281
author Michalis Spyrou <michalis.spyrou@arm.com> Thu Nov 22 17:36:28 2018 +0000
committer Michalis Spyrou <michalis.spyrou@arm.com> Fri Nov 30 15:46:49 2018 +0000
tree d17899ba82203423320bfa8d2dea1e07b045c898
parent 95abfddfa08ab85d4f88c6f4d2e077969178f2d5

COMPMID-1728 CL: Implement ArgMax/ArgMin

Change-Id: I7eae2e55cc0b0b7bbebb7617299daaca6f75f40c
Reviewed-on: https://review.mlplatform.org/292
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

arm_compute/runtime/CL/CLFunctions.h

diff --git a/arm_compute/runtime/CL/CLFunctions.h b/arm_compute/runtime/CL/CLFunctions.h
index d9b29ff..780597e 100644
--- a/arm_compute/runtime/CL/CLFunctions.h
+++ b/arm_compute/runtime/CL/CLFunctions.h
@@ -28,6 +28,7 @@
 #include "arm_compute/runtime/CL/functions/CLAbsoluteDifference.h"
 #include "arm_compute/runtime/CL/functions/CLAccumulate.h"
 #include "arm_compute/runtime/CL/functions/CLActivationLayer.h"
+#include "arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h"
 #include "arm_compute/runtime/CL/functions/CLArithmeticAddition.h"
 #include "arm_compute/runtime/CL/functions/CLArithmeticDivision.h"
 #include "arm_compute/runtime/CL/functions/CLArithmeticSubtraction.h"