Update TensorFlow Lite to v2.6.0. Fix build for Android with NNAPI

iwatake2222 · Aug 16, 2021 · f376227 · f376227
1 parent 0c3ad42
commit f376227
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/inference_helper/inference_helper_tensorflow_lite.cpp b/inference_helper/inference_helper_tensorflow_lite.cpp
@@ -40,6 +40,10 @@ limitations under the License.
 #include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
 #endif
 
+#ifdef INFERENCE_HELPER_ENABLE_TFLITE_DELEGATE_NNAPI
+#include "tensorflow/lite/delegates/nnapi/nnapi_delegate.h"
+#endif
+
 /* for My modules */
 #include "inference_helper_log.h"
 #include "inference_helper_tensorflow_lite.h"