summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMiao Wang <miaowang@google.com>2019-01-29 13:08:44 -0800
committerandroid-build-merger <android-build-merger@google.com>2019-01-29 13:08:44 -0800
commit03c46a4e0d6e13c2dd85185f4dd5c8aa892c5078 (patch)
tree636bcc59c21dae57b05f173d4d38f159d860ecc3
parent3423fa762195af108e6099456f850055293d2940 (diff)
parent537cdd140893e1078d8e13754a2ede92617f4aff (diff)
downloadlibtextclassifier-03c46a4e0d6e13c2dd85185f4dd5c8aa892c5078.tar.gz
Update include dir after TFLite rebase. am: db61aac008 am: f6dd76cb57
am: 537cdd1408 Change-Id: Ie763ab095ff9e135f248a144c2fbaabba2790a49
-rw-r--r--utils/tflite-model-executor.cc2
-rw-r--r--utils/tflite-model-executor.h10
2 files changed, 6 insertions, 6 deletions
diff --git a/utils/tflite-model-executor.cc b/utils/tflite-model-executor.cc
index 05762d0..4877d4a 100644
--- a/utils/tflite-model-executor.cc
+++ b/utils/tflite-model-executor.cc
@@ -16,8 +16,8 @@
#include "utils/tflite-model-executor.h"
+#include "tensorflow/lite/kernels/register.h"
#include "utils/base/logging.h"
-#include "tensorflow/contrib/lite/kernels/register.h"
// Forward declaration of custom TensorFlow Lite ops for registration.
namespace tflite {
diff --git a/utils/tflite-model-executor.h b/utils/tflite-model-executor.h
index fd00924..ab1e76d 100644
--- a/utils/tflite-model-executor.h
+++ b/utils/tflite-model-executor.h
@@ -21,13 +21,13 @@
#include <memory>
+#include "tensorflow/lite/interpreter.h"
+#include "tensorflow/lite/kernels/register.h"
+#include "tensorflow/lite/model.h"
+#include "tensorflow/lite/op_resolver.h"
+#include "tensorflow/lite/string_util.h"
#include "utils/base/logging.h"
#include "utils/tensor-view.h"
-#include "tensorflow/contrib/lite/interpreter.h"
-#include "tensorflow/contrib/lite/kernels/register.h"
-#include "tensorflow/contrib/lite/model.h"
-#include "tensorflow/contrib/lite/op_resolver.h"
-#include "tensorflow/contrib/lite/string_util.h"
namespace libtextclassifier3 {