aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/Conv3D.h
diff options
context:
space:
mode:
authorGiorgio Arena <giorgio.arena@arm.com>2021-10-19 15:45:57 +0100
committerSheri Zhang <sheri.zhang@arm.com>2021-10-20 15:54:24 +0000
commit51847d5dd9cad6bc81673642a01fd531def44311 (patch)
tree1c5b79334d054141308c1e03e05749ac33ed5d34 /tests/validation/reference/Conv3D.h
parent7a8cf1707e45ea011e0db5c0b3091c381ccd387f (diff)
downloadComputeLibrary-51847d5dd9cad6bc81673642a01fd531def44311.tar.gz
Implement CLDirectConv3DKernel - uint8/int8
Resolve COMPMID-4663 Signed-off-by: Giorgio Arena <giorgio.arena@arm.com> Change-Id: I5c3c1cffed5385c06b789543318f7f4d6096987e Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6468 Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Diffstat (limited to 'tests/validation/reference/Conv3D.h')
-rw-r--r--tests/validation/reference/Conv3D.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/validation/reference/Conv3D.h b/tests/validation/reference/Conv3D.h
index ade8a2c242..e3674f4bfb 100644
--- a/tests/validation/reference/Conv3D.h
+++ b/tests/validation/reference/Conv3D.h
@@ -37,8 +37,8 @@ namespace validation
{
namespace reference
{
-template <typename T>
-SimpleTensor<T> conv3d(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, const SimpleTensor<T> &bias, SimpleTensor<T> &dst,
+template <typename T, typename TB>
+SimpleTensor<T> conv3d(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, const SimpleTensor<TB> &bias, SimpleTensor<T> &dst,
const Conv3dInfo &conv3d_info);
} // namespace reference
} // namespace validation