| commit | a78f829a923ef16a65d0165fde5dedbaced21c84 | [log] [tgz] |
|---|---|---|
| author | Xusong Wang <xusongw@google.com> | Wed Sep 26 13:25:26 2018 -0700 |
| committer | Xusong Wang <xusongw@google.com> | Mon Oct 15 16:14:42 2018 -0700 |
| tree | 60d2420dad35202acbdcc500f61ba751f1a7d5ab | |
| parent | 6a362bae1e339bc91d208054f630bd3c0cc98334 [diff] |
Add NCHW data layout support to DEPTH_TO_SPACE. Add an optional parameter data_layout to DEPTH_TO_SPACE op for NCHW data layout support. Create testcases and generate test. Provide reference CPU implementation for NCHW support. The native layout for CPU implementation is still NHWC. Appropriate transpositions are applied for NCHW input/output. Bug: 112320705 Test: NeuralNetworksTest_static Change-Id: Ifa383bac6d7802d991587504eb60e7ce78e52103