diff options
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. |