diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_constant_folding.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_constant_folding.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_constant_folding.cc b/tensorflow/compiler/xla/service/hlo_constant_folding.cc index 804efdd906..1a2eed5f60 100644 --- a/tensorflow/compiler/xla/service/hlo_constant_folding.cc +++ b/tensorflow/compiler/xla/service/hlo_constant_folding.cc @@ -42,6 +42,9 @@ StatusOr<bool> HloConstantFolding::Run(HloModule* module) { bool changed = false; for (auto& computation : module->computations()) { + if (computation->IsFusionComputation()) { + continue; + } for (auto instruction : computation->MakeInstructionPostOrder()) { // Skip dead code. if (instruction->user_count() == 0 && |