diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-10 16:09:00 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-10 16:12:07 -0700 |
commit | 874cf8e1d332175c8a90d7512f8385e98e2a7377 (patch) | |
tree | 129149c6c73e828246b7202dccdfb4d9ce583de3 /tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc | |
parent | 66b6dda1b77cbf075e94009718446511fa13dd41 (diff) |
Enable support for crops in BatchToSpaceNd
PiperOrigin-RevId: 196186750
Diffstat (limited to 'tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc b/tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc index 8485cde1b4..95b025c1b3 100644 --- a/tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc +++ b/tensorflow/contrib/lite/kernels/batch_to_space_nd_test.cc @@ -120,16 +120,16 @@ TEST(BatchToSpaceNDOpTest, InvalidShapeTest) { } TEST(BatchToSpaceNDOpTest, InvalidCropsConstTest) { - EXPECT_DEATH(BatchToSpaceNDOpConstModel({3, 2, 2, 1}, {2, 2}, {0, 0, 0, 1}), - "1 != 0"); + EXPECT_DEATH(BatchToSpaceNDOpConstModel({3, 2, 2, 1}, {2, 2}, {0, 0, 0, -1}), + "crops.3. >= 0 was not true."); } TEST(BatchToSpaceNDOpTest, InvalidCropsDynamicTest) { BatchToSpaceNDOpDynamicModel m({4, 2, 2, 1}); m.SetInput({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}); m.SetBlockShape({2, 2}); - m.SetCrops({0, 0, 1, 0}); - EXPECT_DEATH(m.Invoke(), "1 != 0"); + m.SetCrops({0, 0, -1, 0}); + EXPECT_DEATH(m.Invoke(), "crops.2. >= 0 was not true."); } } // namespace |