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

Merge pull request #27372 from Dayananda-V:tflite_zeros_like_test_append

Browse Source 
PiperOrigin-RevId: 241412249
This commit is contained in:
TensorFlower Gardener 2019-04-01 15:37:00 -07:00
commit 3c98419e46
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/lite/kernels/zeros_like_test.cc
View File

@ -68,6 +68,22 @@ TEST(ZerosLikeOpModel, ZerosLikeInt64) {
EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 2, 2, 1}));
}
#ifdef GTEST_HAS_DEATH_TEST
TEST(ZerosLikeOpModel, InvalidTypeTest) {
ZerosLikeOpModel m_uint8({TensorType_UINT8, {1, 1}});
EXPECT_DEATH(m_uint8.Invoke(),
"ZerosLike only currently supports int64, int32, and float32");
ZerosLikeOpModel m_int16({TensorType_INT16, {1, 1}});
EXPECT_DEATH(m_int16.Invoke(),
"ZerosLike only currently supports int64, int32, and float32");
ZerosLikeOpModel m_complex({TensorType_COMPLEX64, {1, 1}});
EXPECT_DEATH(m_complex.Invoke(),
"ZerosLike only currently supports int64, int32, and float32");
ZerosLikeOpModel m_int8({TensorType_INT8, {1, 1}});
EXPECT_DEATH(m_int8.Invoke(),
"ZerosLike only currently supports int64, int32, and float32");
}
#endif
} // namespace
} // namespace tflite