From e1dcf1b5934d4feeb0aa30fa625f77d4a47283ef Mon Sep 17 00:00:00 2001 From: Nicoletta <144339210+Nico-letta@users.noreply.github.com> Date: Mon, 22 Jun 2026 17:27:31 +0100 Subject: [PATCH] Exclude 'Reciprocal' from Involution check --- tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,