diff options
| author |  Yuanzhong Xu <yuanzx@google.com> | 2018-02-26 10:42:59 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-26 10:50:19 -0800 |
| commit | ca328de4d8805a7495485e787811484d843c43a2 (patch) | |
| tree | ff5046554578b45a4db835f4d5fe099a2ded4ec0 | |
| parent | 0b94d6270866789d210d1914e60937b6f231a669 (diff) | |
[XLA] Add kConvert to EffectiveOperandPrecisionIsOutputPrecision list.
PiperOrigin-RevId: 187044921
| -rw-r--r-- | tensorflow/compiler/xla/service/bfloat16_support.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/bfloat16_support.cc b/tensorflow/compiler/xla/service/bfloat16_support.cc index 3fd9e24601..07b4b14b5e 100644 --- a/tensorflow/compiler/xla/service/bfloat16_support.cc +++ b/tensorflow/compiler/xla/service/bfloat16_support.cc @@ -79,6 +79,7 @@ bool BFloat16Support::EffectiveOperandPrecisionIsOutputPrecision( case HloOpcode::kBroadcast: case HloOpcode::kClamp: case HloOpcode::kConcatenate: + case HloOpcode::kConvert: case HloOpcode::kCopy: case HloOpcode::kGetTupleElement: case HloOpcode::kMaximum: |