diff options
author | 2018-08-29 17:48:09 -0700 | |
---|---|---|
committer | 2018-08-29 17:51:38 -0700 | |
commit | 7cda8c3a8ad528f2e11fc47b0abf08e01f97af45 (patch) | |
tree | 4abceb5e1c3ca6692f41d53d71f8e4de1c4108fb /tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc | |
parent | e528493c8cde468451ba1b1995e649ebe9c29b02 (diff) |
[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
Diffstat (limited to 'tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc | 1 |
1 files changed, 1 insertions, 0 deletions
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: |