Automated g4 rollback of changelist 192180356

PiperOrigin-RevId: 193427566

1 files changed, 2 insertions, 5 deletions

diff --git a/tensorflow/compiler/xla/service/instruction_fusion.cc b/tensorflow/compiler/xla/service/instruction_fusion.cc
index 3f4dbf897d..d69ad80bdb 100644
--- a/tensorflow/compiler/xla/service/instruction_fusion.cc
+++ b/tensorflow/compiler/xla/service/instruction_fusion.cc
@@ -37,7 +37,6 @@ namespace xla {
     case HloOpcode::kBitcast:
     case HloOpcode::kBitcastConvert:
     case HloOpcode::kBroadcast:
-    case HloOpcode::kBroadcastDimOne:
     case HloOpcode::kCeil:
     case HloOpcode::kClamp:
     case HloOpcode::kComplex:
@@ -143,8 +142,7 @@ bool InstructionFusion::EffectivelyUnary(HloInstruction* hlo) {
       });
   return std::count_if(hlo->operands().begin(), hlo->operands().end(),
                        [output_rank](HloInstruction* operand) {
-                         if (operand->opcode() == HloOpcode::kBroadcast ||
-                             operand->opcode() == HloOpcode::kBroadcastDimOne) {
+                         if (operand->opcode() == HloOpcode::kBroadcast) {
                            return false;
                          }
                          if (operand->opcode() == HloOpcode::kConstant &&
@@ -249,8 +247,7 @@ StatusOr<bool> InstructionFusion::Run(HloModule* module) {
     auto reachability = computation->ComputeReachability();
 
     auto cheap_to_duplicate = [this](HloInstruction* producer) {
-      if (producer->opcode() == HloOpcode::kBroadcast ||
-          producer->opcode() == HloOpcode::kBroadcastDimOne) {
+      if (producer->opcode() == HloOpcode::kBroadcast) {
         return true;
       }
       if (producer->opcode() == HloOpcode::kConstant &&