diff options
-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); } |