From 7cda8c3a8ad528f2e11fc47b0abf08e01f97af45 Mon Sep 17 00:00:00 2001 From: David Majnemer Date: Wed, 29 Aug 2018 17:48:09 -0700 Subject: [XLA] Switch to using kIota from TF We were using a broadcast of a constant instead of the kIota HLO. To make switching to kIota practical, we need to do a few things first: - Don't constant fold kIota. - Don't hoist kIota from loops without good cause. PiperOrigin-RevId: 210825834 --- tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc | 1 + 1 file changed, 1 insertion(+) (limited to 'tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc') diff --git a/tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc b/tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc index f4098f28b3..e8fe33e626 100644 --- a/tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc +++ b/tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc @@ -110,6 +110,7 @@ bool WhileLoopInvariantCodeMotion::NotWorthHoistingIndividually( case HloOpcode::kBitcast: case HloOpcode::kBroadcast: + case HloOpcode::kIota: case HloOpcode::kReshape: case HloOpcode::kReverse: case HloOpcode::kSlice: -- cgit v1.2.3