From 47370943471c98b5ed4c954b350b925d368a810e Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Mon, 13 Nov 2023 17:20:45 +0000 Subject: Optimize CPU depth-to-space Resolves: COMPMID-6622 Signed-off-by: Viet-Hoa Do Change-Id: Ibac276618bdda125dcbb9c851c547f12739b15b4 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10749 Tested-by: Arm Jenkins Reviewed-by: Gunes Bayir Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- src/BUILD.bazel | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/BUILD.bazel') diff --git a/src/BUILD.bazel b/src/BUILD.bazel index c14e10c836..f0c4b52688 100644 --- a/src/BUILD.bazel +++ b/src/BUILD.bazel @@ -733,6 +733,8 @@ filegroup( "cpu/kernels/crop/generic/neon/fp16.cpp", "cpu/kernels/crop/generic/neon/fp32.cpp", "cpu/kernels/crop/generic/neon/integer.cpp", + "cpu/kernels/depth_to_space/nchw/any/impl.cpp", + "cpu/kernels/depth_to_space/nhwc/any/impl.cpp", "cpu/kernels/depthwiseconv2d/generic/neon/fp16.cpp", "cpu/kernels/depthwiseconv2d/generic/neon/fp32.cpp", "cpu/kernels/depthwiseconv2d/generic/neon/impl.cpp", -- cgit v1.2.1