From 2fea13593a4753316ae488edf489cb4b00150153 Mon Sep 17 00:00:00 2001 From: Mohammed Suhail Munshi Date: Mon, 29 Apr 2024 22:53:58 +0100 Subject: Add batched indices support to Scatter GPU Implementation Resolves: [COMPMID-6897] Signed-off-by: Mohammed Suhail Munshi Change-Id: I70b1c3c5f0de8484fcb6c3b0cc0d0d8c059b0f58 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11525 Comments-Addressed: Arm Jenkins Reviewed-by: Gunes Bayir Tested-by: Arm Jenkins Benchmark: Arm Jenkins --- tests/validation/reference/ScatterLayer.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'tests/validation/reference') diff --git a/tests/validation/reference/ScatterLayer.cpp b/tests/validation/reference/ScatterLayer.cpp index c9e6035e14..55c48a9002 100644 --- a/tests/validation/reference/ScatterLayer.cpp +++ b/tests/validation/reference/ScatterLayer.cpp @@ -63,6 +63,7 @@ T reduce_op(const T ¤t,const T &update,const ScatterFunction func) } template float reduce_op(const float ¤t,const float &update,const ScatterFunction func); +template half reduce_op(const half ¤t,const half &update,const ScatterFunction func); } // NOTE: This function expects collapsed tensors as input. -- cgit v1.2.1