diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc index 5802e437683eb1..74e49cbb17f76d 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc @@ -1309,7 +1309,7 @@ class RemoveInvolution : public ArithmeticOptimizerStage { ~RemoveInvolution() override = default; bool IsSupported(const NodeDef* node) const override { - return IsInvolution(*node); + return IsInvolution(*node) && node->op() != "Reciprocal"; } absl::Status TrySimplify(NodeDef* node,