commit 569d6605c816384e3acb265d97c884577e6b377c
author Xusong Wang <xusongw@google.com> Mon Oct 15 17:40:16 2018 -0700
committer Xusong Wang <xusongw@google.com> Mon Nov 05 15:05:21 2018 -0800
tree 8f92243022cb7ded09a2d6550b190135b96b3fec
parent f067a0fd1238de6d64bd225997f5085122865c9c

Add NCHW data layout support to GROUPED_CONV_2D.

Add a parameter data_layout to GROUPED_CONV_2D 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: 113562731

Test: NeuralNetworksTest_static
Change-Id: Ic2a4b2ac27fdd5bead7721a563fdddc43bd85aac
Merged-In: Ic2a4b2ac27fdd5bead7721a563fdddc43bd85aac
(cherry picked from commit c510d397d608c94351911eca07521d3211a537bc)