diff options
author | 2018-08-17 22:16:36 -0700 | |
---|---|---|
committer | 2018-08-17 22:20:29 -0700 | |
commit | 39b17f9d56d6b9a02a30bfa24ac9e15ab37ca761 (patch) | |
tree | f6ded73c971ab178bde3f1c319e6ce867e95ae29 /tensorflow/compiler/xla/service/while_loop_constant_sinking.cc | |
parent | 4a41f50648929197954d892559587cb76458d306 (diff) |
Automated rollback of commit 4a41f50648929197954d892559587cb76458d306
PiperOrigin-RevId: 209248552
Diffstat (limited to 'tensorflow/compiler/xla/service/while_loop_constant_sinking.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/while_loop_constant_sinking.cc | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/tensorflow/compiler/xla/service/while_loop_constant_sinking.cc b/tensorflow/compiler/xla/service/while_loop_constant_sinking.cc index aab1180662..62af45128a 100644 --- a/tensorflow/compiler/xla/service/while_loop_constant_sinking.cc +++ b/tensorflow/compiler/xla/service/while_loop_constant_sinking.cc @@ -14,7 +14,6 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/compiler/xla/service/while_loop_constant_sinking.h" -#include "absl/algorithm/container.h" #include "tensorflow/compiler/xla/service/while_util.h" #include "tensorflow/compiler/xla/util.h" #include "tensorflow/core/lib/gtl/flatmap.h" @@ -33,7 +32,7 @@ static Status ReplaceUsesWhileKeepingLoopInvariance( std::vector<HloInstruction*> users; users.reserve(old_instr->user_count()); - absl::c_copy(old_instr->users(), std::back_inserter(users)); + c_copy(old_instr->users(), std::back_inserter(users)); for (auto* user : users) { for (int64 i = 0, e = user->operand_count(); i < e; i++) { @@ -109,10 +108,10 @@ StatusOr<bool> WhileLoopConstantSinking::Run(HloModule* module) { // // This will let us sink the constant into the outer while first and then // into the inner while in a single run of this pass. - absl::c_copy_if(comp->instructions(), std::back_inserter(while_instrs), - [](const HloInstruction* instr) { - return instr->opcode() == HloOpcode::kWhile; - }); + c_copy_if(comp->instructions(), std::back_inserter(while_instrs), + [](const HloInstruction* instr) { + return instr->opcode() == HloOpcode::kWhile; + }); } for (HloInstruction* while_instr : while_instrs) { |