From d6afedc775220f17317f1835a4d18b72a54525de Mon Sep 17 00:00:00 2001 From: Chunosov Date: Mon, 6 Nov 2017 22:09:45 +0700 Subject: COMPMID-661: softmax-fp32 optimisation (#14) Change-Id: I2007af1ed9dcf68065cf412aa50f73a2025b31a6 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/94605 Reviewed-by: Gian Marco Iodice Tested-by: Kaizen --- src/core/CL/CLKernelLibrary.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/core/CL/CLKernelLibrary.cpp') diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 32199525b0..6efeebd63f 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -300,6 +300,8 @@ const std::map CLKernelLibrary::_kernel_program_map = { "softmax_layer_max", "softmax_layer.cl" }, { "softmax_layer_shift_exp_sum", "softmax_layer.cl" }, { "softmax_layer_norm", "softmax_layer.cl" }, + { "softmax_layer_max_shift_exp_sum_serial", "softmax_layer.cl" }, + { "softmax_layer_max_shift_exp_sum_parallel", "softmax_layer.cl" }, { "suppress_non_maximum", "canny.cl" }, { "tablelookup_U8", "tablelookup.cl" }, { "tablelookup_S16", "tablelookup.cl" }, -- cgit v1.2.1