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

Add fix for lsh_projection_test

Browse Source
This commit is contained in:
Sidong-Wei 2020-11-19 11:54:12 -05:00
parent 0305e6d887
commit 8c87484fc0
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/lite/kernels/lsh_projection_test.cc
View File

@ -87,7 +87,12 @@ TEST(LSHProjectionOpTest2, Dense1DInputs) {
m.Invoke();
#if defined(__s390x__)
// Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 1, 1, 1, 0));
#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 0, 1, 0, 0));
#endif
}
TEST(LSHProjectionOpTest2, Sparse1DInputs) {
@ -98,7 +103,12 @@ TEST(LSHProjectionOpTest2, Sparse1DInputs) {
m.Invoke();
#if defined(__s390x__)
// Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 1, 8 + 0));
#endif
}
TEST(LSHProjectionOpTest2, Sparse3DInputs) {
@ -111,7 +121,12 @@ TEST(LSHProjectionOpTest2, Sparse3DInputs) {
m.Invoke();
#if defined(__s390x__)
// Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 2, 4 + 1, 8 + 1));
#endif
}
} // namespace