diff options
| author |  David Majnemer <majnemer@google.com> | 2018-08-29 18:42:05 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-29 18:45:49 -0700 |
| commit | fb65d4ce8dbc06452df11891facc654076891488 (patch) | |
| tree | 5378feafef2302d75cce03852865c2aa951966df /tensorflow/compiler/xla/service/cpu | |
| parent | bea2cb338c268eb25e32d929772dfe6c99f713bd (diff) | |
[XLA] Enable fusing kIota for CPU & GPU
PiperOrigin-RevId: 210832078
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu')
| -rw-r--r-- | tensorflow/compiler/xla/service/cpu/cpu_instruction_fusion.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/cpu_instruction_fusion.cc b/tensorflow/compiler/xla/service/cpu/cpu_instruction_fusion.cc index 7f867fa149..f9cd61bea3 100644 --- a/tensorflow/compiler/xla/service/cpu/cpu_instruction_fusion.cc +++ b/tensorflow/compiler/xla/service/cpu/cpu_instruction_fusion.cc @@ -35,7 +35,7 @@ bool CanBeLoopFused(const HloInstruction& hlo) { hlo.opcode() == HloOpcode::kDynamicSlice || hlo.opcode() == HloOpcode::kDynamicUpdateSlice || hlo.opcode() == HloOpcode::kGather || - hlo.opcode() == HloOpcode::kPad || + hlo.opcode() == HloOpcode::kIota || hlo.opcode() == HloOpcode::kPad || hlo.opcode() == HloOpcode::kReshape || hlo.opcode() == HloOpcode::kReverse || hlo.opcode() == HloOpcode::kSlice || |