commit c1f6b09cc22b6e2de3d9eb88aec1778d0308a2b3
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon May 11 16:10:38 2020 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed May 13 18:34:12 2020 +0000
tree 14a70415b6d17b42246798a293dfdd13cbe4c123
parent 0c32ccfcd0b770cdd4eeb9d778b7f72a233e229a

IVGCVSW-4753 Refactor CL Softmax workload generalizing for different datatype

    * Change ComputeSoftmaxAclAxis to work with int and uint axis


Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ibbfa9ec7e2f0416e6885673212a767419c871cca

src/backends/cl/workloads/ClWorkloads.hpp

diff --git a/src/backends/cl/workloads/ClWorkloads.hpp b/src/backends/cl/workloads/ClWorkloads.hpp
index ffe66a0..62b73da 100644
--- a/src/backends/cl/workloads/ClWorkloads.hpp
+++ b/src/backends/cl/workloads/ClWorkloads.hpp
@@ -39,8 +39,7 @@
 #include "ClResizeWorkload.hpp"
 #include "ClRsqrtWorkload.hpp"
 #include "ClSliceWorkload.hpp"
-#include "ClSoftmaxFloatWorkload.hpp"
-#include "ClSoftmaxUint8Workload.hpp"
+#include "ClSoftmaxWorkload.hpp"
 #include "ClSpaceToBatchNdWorkload.hpp"
 #include "ClSpaceToDepthWorkload.hpp"
 #include "ClSplitterWorkload.hpp"