diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-09 15:58:28 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-09 16:02:11 -0700 |
commit | f1badb6664c290176864d1a1d4ab537b7332b730 (patch) | |
tree | b029fd120467c8ec9a331b576b7fc755a6c207a2 /tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | |
parent | ef58a46b730155717f1b03abb20767c1924ad05e (diff) |
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
Diffstat (limited to 'tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc')
-rw-r--r-- | tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc index adfae2e1a3..f46c30c92c 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc @@ -2233,6 +2233,9 @@ string ArithmeticOptimizer::TrySimplifyAndReplaceUses( new_square_node->set_input(i - 1, new_square_node->input(i)); } new_square_node->mutable_input()->RemoveLast(); + for (const string& input : new_square_node->input()) { + node_map_->AddOutput(NodeName(input), new_square_node->name()); + } return new_square_node->name(); } } |