aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/BUILD
diff options
context:
space:
mode:
authorGravatar Yu-Cheng Ling <ycling@google.com>2018-05-13 19:52:18 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-13 19:55:02 -0700
commit699b217cd6c5ddc0832be8471dde47999829e435 (patch)
tree035167be1ec270dded665347d20ec9385bed0fcc /tensorflow/contrib/lite/BUILD
parent2fbc0c5a45955c877e0a165bb561fc2f01518321 (diff)
Introduce op version into TFLite
PiperOrigin-RevId: 196448769
Diffstat (limited to 'tensorflow/contrib/lite/BUILD')
-rw-r--r--tensorflow/contrib/lite/BUILD14
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/BUILD b/tensorflow/contrib/lite/BUILD
index 10065e894c..01c76b7a66 100644
--- a/tensorflow/contrib/lite/BUILD
+++ b/tensorflow/contrib/lite/BUILD
@@ -114,6 +114,7 @@ cc_library(
"interpreter.cc",
"model.cc",
"nnapi_delegate.cc",
+ "op_resolver.cc",
"optional_debug_tools.cc",
],
hdrs = [
@@ -124,6 +125,7 @@ cc_library(
"interpreter.h",
"model.h",
"nnapi_delegate.h",
+ "op_resolver.h",
"optional_debug_tools.h",
],
copts = tflite_copts(),
@@ -226,6 +228,18 @@ cc_test(
],
)
+# Test OpResolver.
+cc_test(
+ name = "op_resolver_test",
+ size = "small",
+ srcs = ["op_resolver_test.cc"],
+ deps = [
+ ":framework",
+ "//tensorflow/contrib/lite/testing:util",
+ "@com_google_googletest//:gtest",
+ ],
+)
+
# Test the C extension API code.
cc_test(
name = "context_test",