Skip to content

Commit

Permalink
[XLA] Fix parameter type of IsLiteralWithValue in algebraic_simplifie…
Browse files Browse the repository at this point in the history
…r.cc.

It should match the param type of LiteralUtil::IsAll.
Change: 146692488
  • Loading branch information
Justin Lebar authored and tensorflower-gardener committed Feb 6, 2017
1 parent f798ce6 commit 06429f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow/compiler/xla/service/algebraic_simplifier.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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);
}
Expand Down

0 comments on commit 06429f5

Please sign in to comment.