From 0d523cccb549e4ff9dd231d033d612391ca31c85 Mon Sep 17 00:00:00 2001 From: steniu01 Date: Thu, 13 Jul 2017 14:24:23 +0100 Subject: COMPMID-443 Change CLSoftMaxLayerKernel to use 3D tensor and collapse the higer dimension Change-Id: I730ef45d855113d8baa7d89818441e168ea43c63 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80573 Reviewed-by: Georgios Pinitas Tested-by: Kaizen Reviewed-by: Anthony Barbier --- arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h') diff --git a/arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h b/arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h index 137b60b25f..5c0d05f586 100644 --- a/arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h +++ b/arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h @@ -24,14 +24,14 @@ #ifndef __ARM_COMPUTE_CLSOFTMAXLAYERKERNEL_H__ #define __ARM_COMPUTE_CLSOFTMAXLAYERKERNEL_H__ -#include "arm_compute/core/CL/ICLSimple2DKernel.h" +#include "arm_compute/core/CL/ICLSimple3DKernel.h" namespace arm_compute { class ICLTensor; /** Interface for the identifying the max value of 1D Logits */ -class CLLogits1DMaxKernel : public ICLSimple2DKernel +class CLLogits1DMaxKernel : public ICLSimple3DKernel { public: /** Set the input and output tensors. -- cgit v1.2.1