commit bbe85068bb7aa6249a4e915462014016373c945f
author Artsiom Ablavatski <artsiom@google.com> Thu Nov 05 14:07:37 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Thu Nov 05 14:08:19 2020 -0800
tree a3c4ce9a0c6250c96abd5f17e01c5f9e61e1d564
parent e784186c457fce6108d030fc9c2ccf89083cf6a7

Introduce DEPTH_TO_SPACE operator and enable it for graph rewriting

PiperOrigin-RevId: 340925373

CMakeLists.txt

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0fe83cc..c5c07c9 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -166,6 +166,7 @@
   src/operators/convolution-nchw.c
   src/operators/convolution-nhwc.c
   src/operators/deconvolution-nhwc.c
+  src/operators/depth-to-space-chw2hwc.c
   src/operators/fully-connected-nc.c
   src/operators/global-average-pooling-ncw.c
   src/operators/global-average-pooling-nwc.c
@@ -190,6 +191,7 @@
   src/subgraph/convolution-2d.c
   src/subgraph/deconvolution-2d.c
   src/subgraph/depthwise-convolution-2d.c
+  src/subgraph/depth-to-space.c
   src/subgraph/divide.c
   src/subgraph/floor.c
   src/subgraph/fully-connected.c