Simplify the remapper code and added support for non scalar mean, variance, scale...
authorBenoit Steiner <bsteiner@google.com>
Wed, 23 May 2018 23:45:26 +0000 (16:45 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 23 May 2018 23:48:01 +0000 (16:48 -0700)
commitdac1f124020234fe24e8893a981b15395d0c6de8
tree54b3b50356fba1d4446f43a4fc50cd1202e21828
parent4ba9e8eed9dfe0727db000bdd8be5384f39e6bd9
Simplify the remapper code and added support for non scalar mean, variance, scale and offset.

PiperOrigin-RevId: 197812268
tensorflow/core/grappler/optimizers/BUILD
tensorflow/core/grappler/optimizers/remapper.cc
tensorflow/core/grappler/optimizers/remapper_test.cc
tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc