commit 0e521177bb539c2d6f0ac96923684975e59a1862
author Marat Dukhan <maratek@google.com> Wed Nov 25 13:10:04 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Nov 25 13:10:41 2020 -0800
tree c1759be9cc2581c2d757d87c259a3659ce8eb556
parent 1810c6eb34006c39a7aa44a7f5d7c01b8399cef9

NHWC version of Depth-To-Space operator

PiperOrigin-RevId: 344304885

CMakeLists.txt

diff --git a/CMakeLists.txt b/CMakeLists.txt
index dbb8711..b0168bc 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -167,6 +167,7 @@
   src/operators/convolution-nhwc.c
   src/operators/deconvolution-nhwc.c
   src/operators/depth-to-space-nchw2nhwc.c
+  src/operators/depth-to-space-nhwc.c
   src/operators/fully-connected-nc.c
   src/operators/global-average-pooling-ncw.c
   src/operators/global-average-pooling-nwc.c
@@ -2929,6 +2930,15 @@
   TARGET_LINK_LIBRARIES(depth-to-space-nchw2nhwc-test PRIVATE XNNPACK gtest gtest_main)
   ADD_TEST(depth-to-space-nchw2nhwc-test depth-to-space-nchw2nhwc-test)
 
+  ADD_EXECUTABLE(depth-to-space-nhwc-test test/depth-to-space-nhwc.cc)
+  SET_TARGET_PROPERTIES(depth-to-space-nhwc-test PROPERTIES
+    CXX_STANDARD 11
+    CXX_STANDARD_REQUIRED YES
+    CXX_EXTENSIONS NO)
+  TARGET_INCLUDE_DIRECTORIES(depth-to-space-nhwc-test PRIVATE src test)
+  TARGET_LINK_LIBRARIES(depth-to-space-nhwc-test PRIVATE XNNPACK gtest gtest_main)
+  ADD_TEST(depth-to-space-nhwc-test depth-to-space-nhwc-test)
+
   ADD_EXECUTABLE(divide-nd-test test/divide-nd.cc)
   SET_TARGET_PROPERTIES(divide-nd-test PROPERTIES
     CXX_STANDARD 11