From b09d05b6b198c0fa205005077a094863e4a2a3e8 Mon Sep 17 00:00:00 2001
From: "Jeffrey A. Dean" <jeff@google.com>
Date: Thu, 11 Jun 2020 09:18:41 -0700
Subject: [PATCH] Avoid one redundant call to IsReachable.

PiperOrigin-RevId: 315912341
Change-Id: I1b576cc9afecf1c7aa7f1c1ecef23a0017b7c92e
---
 tensorflow/compiler/xla/service/multi_output_fusion.cc | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tensorflow/compiler/xla/service/multi_output_fusion.cc b/tensorflow/compiler/xla/service/multi_output_fusion.cc
index d97893b6d04..7b4819d8c03 100644
--- a/tensorflow/compiler/xla/service/multi_output_fusion.cc
+++ b/tensorflow/compiler/xla/service/multi_output_fusion.cc
@@ -350,12 +350,13 @@ void MultiOutputFusion::UpdateReachability(
     if (skip != nullptr && skip(instr)) {
       continue;
     }
-    if (reachability_->IsReachable(instr2, instr) &&
-        reachability_->IsReachable(instr1, instr)) {
+    bool instr2_instr = reachability_->IsReachable(instr2, instr);
+    bool instr1_instr = reachability_->IsReachable(instr1, instr);
+    if (instr2_instr && instr1_instr) {
       // If a candidate was already reachable by both, no update needed.
       continue;
     }
-    if (reachability_->IsReachable(instr2, instr)) {
+    if (instr2_instr) {
       reachability_->FastSetReachabilityToUnion({instr, instr1}, instr);
     }
     if (reachability_->IsReachable(instr1, instr)) {