tree 8f92243022cb7ded09a2d6550b190135b96b3fec
parent f067a0fd1238de6d64bd225997f5085122865c9c
author Xusong Wang <xusongw@google.com> 1539650416 -0700
committer Xusong Wang <xusongw@google.com> 1541459121 -0800

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)