Add operator level tests for DEPTH_TO_SPACE op
PiperOrigin-RevId: 340971739
diff --git a/CMakeLists.txt b/CMakeLists.txt
index c5c07c9..9b718ad 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2924,6 +2924,15 @@
TARGET_LINK_LIBRARIES(deconvolution-nhwc-test PRIVATE XNNPACK gtest gtest_main)
ADD_TEST(deconvolution-nhwc-test deconvolution-nhwc-test)
+ ADD_EXECUTABLE(depth-to-space-test test/depth-to-space.cc)
+ SET_TARGET_PROPERTIES(depth-to-space-test PROPERTIES
+ CXX_STANDARD 11
+ CXX_STANDARD_REQUIRED YES
+ CXX_EXTENSIONS NO)
+ TARGET_INCLUDE_DIRECTORIES(depth-to-space-test PRIVATE src test)
+ TARGET_LINK_LIBRARIES(depth-to-space-test PRIVATE XNNPACK gtest gtest_main)
+ ADD_TEST(depth-to-space-test depth-to-space-test)
+
ADD_EXECUTABLE(divide-nd-test test/divide-nd.cc)
SET_TARGET_PROPERTIES(divide-nd-test PROPERTIES
CXX_STANDARD 11