diff options
| author |  Justin Lebar <jlebar@google.com> | 2017-02-06 11:57:56 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-06 12:14:41 -0800 |
| commit | 06429f50497ea709124d240eadd9dd14041b10d2 (patch) | |
| tree | 1a7e6c77c045c9474fdf3bc29274e37cdaa3a949 | |
| parent | f798ce615c1d14badcf945a3d9cc59dae7b35f36 (diff) | |
[XLA] Fix parameter type of IsLiteralWithValue in algebraic_simplifier.cc.
It should match the param type of LiteralUtil::IsAll.
Change: 146692488
| -rw-r--r-- | tensorflow/compiler/xla/service/algebraic_simplifier.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/algebraic_simplifier.cc b/tensorflow/compiler/xla/service/algebraic_simplifier.cc index 0692edc469..d35c6d6adb 100644 --- a/tensorflow/compiler/xla/service/algebraic_simplifier.cc +++ b/tensorflow/compiler/xla/service/algebraic_simplifier.cc @@ -46,7 +46,7 @@ namespace xla { namespace { // Returns whether operand is a literal with the given value. -bool IsLiteralWithValue(const HloInstruction* operand, int value) { +bool IsLiteralWithValue(const HloInstruction* operand, int8 value) { return operand->opcode() == HloOpcode::kConstant && LiteralUtil::IsAll(operand->literal(), value); } |