diff options
author | Nick Desaulniers <ndesaulniers@google.com> | 2018-04-27 16:14:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-27 16:17:22 -0700 |
commit | a52f64de874a0c2624ccdbab4f7b67eea9893e4c (patch) | |
tree | cd29ce866323674b81bad82ac865b1466f4ce601 /tensorflow/compiler/xla/service/hlo_evaluator.cc | |
parent | fbd9ecd7361ff384bc05e30d2b44fc2a1f1da72b (diff) |
[TF:XLA:INTERPRETER] implement bfloat16 comparisons
PiperOrigin-RevId: 194608854
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_evaluator.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_evaluator.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_evaluator.cc b/tensorflow/compiler/xla/service/hlo_evaluator.cc index c5e3014834..f1dcef1dfc 100644 --- a/tensorflow/compiler/xla/service/hlo_evaluator.cc +++ b/tensorflow/compiler/xla/service/hlo_evaluator.cc @@ -2536,6 +2536,11 @@ Status HloEvaluator::HandleCompare(HloInstruction* compare) { } break; case F16: return Unimplemented("unhandled primitive type: F16."); + case BF16: { + TF_ASSIGN_OR_RETURN(evaluated_[compare], + Compare<bfloat16>(compare->shape(), opcode, + lhs_literal, rhs_literal)); + } break; case F32: { TF_ASSIGN_OR_RETURN( evaluated_[compare], |