diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-01 17:33:55 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 17:38:59 -0700 |
commit | bacf1949f92bb1daa9e5c8a31cc6924e532551e9 (patch) | |
tree | cd9756b965a172b96dbbb268ffb27973e6d990e0 /tensorflow/compiler | |
parent | 9a23e9251ecba026471ff77a5bbbc802a2889a10 (diff) |
[XLA] Add kAllToAll and kCollectivePermute to EffectiveOperandPrecisionIsOutputPrecision list.
PiperOrigin-RevId: 215311766
Diffstat (limited to 'tensorflow/compiler')
-rw-r--r-- | tensorflow/compiler/xla/service/bfloat16_support.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/bfloat16_support.cc b/tensorflow/compiler/xla/service/bfloat16_support.cc index 23645346e6..5b48f10505 100644 --- a/tensorflow/compiler/xla/service/bfloat16_support.cc +++ b/tensorflow/compiler/xla/service/bfloat16_support.cc @@ -78,8 +78,10 @@ bool BFloat16Support::EffectiveOperandPrecisionIsOutputPrecision( const HloInstruction& hlo, int64 operand_index) { switch (hlo.opcode()) { case HloOpcode::kAbs: + case HloOpcode::kAllToAll: case HloOpcode::kBroadcast: case HloOpcode::kClamp: + case HloOpcode::kCollectivePermute: case HloOpcode::kConcatenate: case HloOpcode::kConvert: case HloOpcode::kCopy: |