diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/while_loop_simplifier.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/while_loop_simplifier.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/while_loop_simplifier.cc b/tensorflow/compiler/xla/service/while_loop_simplifier.cc index 2fac914892..65734f91bc 100644 --- a/tensorflow/compiler/xla/service/while_loop_simplifier.cc +++ b/tensorflow/compiler/xla/service/while_loop_simplifier.cc @@ -58,9 +58,7 @@ static bool ContainsSendOrRecv(const HloComputation* comp) { static bool IsOrContainsSendOrRecv(const HloInstruction* instr) { if (instr->opcode() == HloOpcode::kSend || - instr->opcode() == HloOpcode::kSendDone || - instr->opcode() == HloOpcode::kRecv || - instr->opcode() == HloOpcode::kRecvDone) { + instr->opcode() == HloOpcode::kRecv) { return true; } for (const auto& subcomp : instr->called_computations()) { |