From 29254aeb11a76c86449c2f38587e9144b2f2aacb Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Fri, 13 Oct 2023 17:40:32 +0100 Subject: Optimize CL softmax * The new softmax implementation consists of only a single kernel. - There are 2 versions of softmax, one for the x dimension and one for any other dimensions. - Softmax kernel handles both native and quantized data type. Resolves: COMPMID-6447 Signed-off-by: Viet-Hoa Do Change-Id: I4a9ae5bc63f78aebeaa85ee48a0d102c9c245eda Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10489 Tested-by: Arm Jenkins Reviewed-by: SiCong Li Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- SConscript | 1 - 1 file changed, 1 deletion(-) (limited to 'SConscript') diff --git a/SConscript b/SConscript index fab9b65acc..8685d01388 100644 --- a/SConscript +++ b/SConscript @@ -451,7 +451,6 @@ if env['opencl'] and env['embed_kernels']: 'src/core/CL/cl_kernels/common/select.cl', 'src/core/CL/cl_kernels/common/slice_ops.cl', 'src/core/CL/cl_kernels/common/softmax_layer.cl', - 'src/core/CL/cl_kernels/common/softmax_layer_quantized.cl', 'src/core/CL/cl_kernels/common/stack_layer.cl', 'src/core/CL/cl_kernels/common/tile.cl', 'src/core/CL/cl_kernels/common/transpose.cl', -- cgit v1.2.1