Add missing update of node map in the Mul(x,x) => Square(x) rewrite. This is what...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Wed, 9 May 2018 22:58:28 +0000 (15:58 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 9 May 2018 23:02:11 +0000 (16:02 -0700)
commitf1badb6664c290176864d1a1d4ab537b7332b730
treeb029fd120467c8ec9a331b576b7fc755a6c207a2
parentef58a46b730155717f1b03abb20767c1924ad05e
Add missing update of node map in the Mul(x,x) => Square(x) rewrite. This is what caused a failure in //photos/vision/object_detection/ranking:brain_embedder_test when the concat/split hoisting was enabled.

PiperOrigin-RevId: 196043455
tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc