diff options
author | steniu01 <steven.niu@arm.com> | 2017-07-13 14:24:23 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | 0d523cccb549e4ff9dd231d033d612391ca31c85 (patch) | |
tree | 9361140af9bb4596db94f62fb07f7151a3c20ccf /arm_compute | |
parent | 00394ae1015c1eaa73f4d98fad31b7771063cd3a (diff) | |
download | ComputeLibrary-0d523cccb549e4ff9dd231d033d612391ca31c85.tar.gz |
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 <georgios.pinitas@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/CL/kernels/CLSoftmaxLayerKernel.h | 4 |
1 files changed, 2 insertions, 2 deletions
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. |