commit 6cbd61f43e08d823f152ffa88016668fadf10848
author ANSHUMAN TRIPATHY <Anshuman.T@huawei.com> Mon Jan 27 11:40:32 2020 +0530
committer ANSHUMAN TRIPATHY <Anshuman.T@huawei.com> Mon Jan 27 11:40:32 2020 +0530
tree 1a37f17a821f119067688e64705fc2dbdd8d72e3
parent cc74783587e59e95040bd05868a67b8b6ae184be

[1] Runtime error for NN API delegates fixed

tensorflow/lite/kernels/batch_to_space_nd_test.cc

diff --git a/tensorflow/lite/kernels/batch_to_space_nd_test.cc b/tensorflow/lite/kernels/batch_to_space_nd_test.cc
index 1b12fde..3ed62db 100644
--- a/tensorflow/lite/kernels/batch_to_space_nd_test.cc
+++ b/tensorflow/lite/kernels/batch_to_space_nd_test.cc
@@ -132,6 +132,11 @@
 }
 
 TEST(BatchToSpaceNDOpTest, SimpleConstTestInt8EmptyOutput) {
+  if (SingleOpModel::GetForceUseNnapi()) {
+    // NNAPI doesn't currently support non-zero crop values.
+    return;
+  }
+
   BatchToSpaceNDOpConstModel m({4, 2, 2, 1}, {2, 2}, {0, 0, 2, 2},
                                TensorType_INT8);
   m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16});
@@ -173,6 +178,11 @@
 }
 
 TEST(BatchToSpaceNDOpTest, SimpleDynamicTestInt8EmptyOutput) {
+  if (SingleOpModel::GetForceUseNnapi()) {
+    // NNAPI doesn't currently support non-zero crop values.
+    return;
+  }
+
   BatchToSpaceNDOpDynamicModel m({4, 2, 2, 1}, TensorType_INT8);
   m.SetInput<int8_t>({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16});
   m.SetBlockShape({2, 2});