[XLA] Fix batchnorm rewriter to not use implicit broadcasts. Algebraic
authorBlake Hechtman <blakehechtman@google.com>
Thu, 31 May 2018 23:58:05 +0000 (16:58 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 1 Jun 2018 00:01:00 +0000 (17:01 -0700)
commit6a6cfbfe4bd79fb0eb21b3d0753d3ddf6ee86ce8
treed45e760fc2015dcf938fba0cc8a05cf257aac699
parent18c67a44ace913d30dc573486dc792300a2cdad3
[XLA] Fix batchnorm rewriter to not use implicit broadcasts. Algebraic
simplifier reshape change is now covered by ReshapeMover.

PiperOrigin-RevId: 198802494
tensorflow/compiler/xla/service/algebraic_simplifier.cc
tensorflow/compiler/xla/service/algebraic_simplifier_test.cc
tensorflow/compiler/xla/service/batchnorm_expander.cc