diff options
| author |  Yuanzhong Xu <yuanzx@google.com> | 2018-04-18 22:07:12 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-18 22:09:02 -0700 |
| commit | ee1676d4dbded64e192aecfa693ab605e24c9929 (patch) | |
| tree | 873690ec656c3f2745956faaf0d83959d6876c87 /tensorflow/compiler/xla/service/bfloat16_propagation.cc | |
| parent | 1d003ee5f82d4d044323a3f162e6cfcf6d645346 (diff) | |
[XLA] Fix BF16 propagation bug for while condition.
PiperOrigin-RevId: 193465140
Diffstat (limited to 'tensorflow/compiler/xla/service/bfloat16_propagation.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/bfloat16_propagation.cc | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/bfloat16_propagation.cc b/tensorflow/compiler/xla/service/bfloat16_propagation.cc index c26d2feef5..43ebe92c5e 100644 --- a/tensorflow/compiler/xla/service/bfloat16_propagation.cc +++ b/tensorflow/compiler/xla/service/bfloat16_propagation.cc @@ -392,7 +392,6 @@ void BFloat16Propagation::AdjustCalledComputationRoot(HloInstruction* hlo) { adjust_computation(hlo->fused_instructions_computation(), hlo->shape()); break; case HloOpcode::kWhile: - adjust_computation(hlo->while_condition(), hlo->shape()); adjust_computation(hlo->while_body(), hlo->shape()); break; default: |