diff options
Diffstat (limited to 'tensorflow/contrib/lite/core/api/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/core/api/BUILD | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/core/api/BUILD b/tensorflow/contrib/lite/core/api/BUILD new file mode 100644 index 0000000000..e4500534f3 --- /dev/null +++ b/tensorflow/contrib/lite/core/api/BUILD @@ -0,0 +1,57 @@ +package( + default_visibility = ["//visibility:public"], +) + +licenses(["notice"]) # Apache 2.0 + +load("//tensorflow/contrib/lite:build_def.bzl", "tflite_copts") + +cc_library( + name = "api", + srcs = [ + "error_reporter.cc", + "flatbuffer_conversions.cc", + "op_resolver.cc", + ], + hdrs = [ + "error_reporter.h", + "flatbuffer_conversions.h", + "op_resolver.h", + ], + copts = tflite_copts(), + deps = [ + "//tensorflow/contrib/lite/c:c_api_internal", + "//tensorflow/contrib/lite/schema:schema_fbs", + ], +) + +cc_test( + name = "error_reporter_test", + size = "small", + srcs = ["error_reporter_test.cc"], + deps = [ + ":api", + "@com_google_googletest//:gtest", + ], +) + +cc_test( + name = "op_resolver_test", + size = "small", + srcs = ["op_resolver_test.cc"], + deps = [ + ":api", + "@com_google_googletest//:gtest", + ], +) + +cc_test( + name = "flatbuffer_conversions_test", + size = "small", + srcs = ["flatbuffer_conversions_test.cc"], + deps = [ + ":api", + "//tensorflow/contrib/lite/c:c_api_internal", + "@com_google_googletest//:gtest", + ], +) |