Support reduction along axis NHW, HW, and C. One use case of "along NHW" is (unfused...
authorYao Zhang <yaozhang@google.com>
Sat, 6 Jan 2018 00:35:41 +0000 (16:35 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 6 Jan 2018 00:39:29 +0000 (16:39 -0800)
commit9f462b27e6df975b1e402c9926d740dfdb9f977a
tree4c1ef5f8f01fa288d14f4c2e18a4f62811ad003e
parent7cb38846cbd62babf2fd3c06959b1bd5e806a7c2
Support reduction along axis NHW, HW, and C. One use case of "along NHW" is (unfused) batch norm. Minor code simplification for squeeze, as added transpose node has shape info now.

PiperOrigin-RevId: 180991050
tensorflow/core/grappler/optimizers/layout_optimizer.cc
tensorflow/python/grappler/layout_optimizer_test.py