experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update literal.cc

Browse Source
This commit is contained in:
tg-at-google 2020-07-29 17:20:29 -04:00 committed by GitHub
parent 3451ef8a16
commit 2158b9de8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/xla/literal.cc
View File

@ -1894,13 +1894,13 @@ bool LiteralBase::IsR1Iota() const {
auto is_iota_at_idx = [&](const int64 idx) {
switch (shape().element_type()) {
case U8:
return static_cast<uint8>(Get<uint8>({idx})) == idx;
return static_cast<int64>(Get<uint8>({idx})) == idx;
case U16:
return static_cast<uint16>(Get<uint16>({idx})) == idx;
return static_cast<int64>(Get<uint16>({idx})) == idx;
case U32:
return static_cast<uint32>(Get<uint32>({idx})) == idx;
return static_cast<int64>(Get<uint32>({idx})) == idx;
case U64:
return static_cast<uint64>(Get<uint64>({idx})) == idx;
return static_cast<int64>(Get<uint64>({idx})) == idx;
case S8:
return Get<int8>({idx}) == idx;
case S16: