diff options
Diffstat (limited to 'tensorflow/contrib/lite/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/BUILD | 46 |
1 files changed, 20 insertions, 26 deletions
diff --git a/tensorflow/contrib/lite/BUILD b/tensorflow/contrib/lite/BUILD index 0091587bf7..f320b53d94 100644 --- a/tensorflow/contrib/lite/BUILD +++ b/tensorflow/contrib/lite/BUILD @@ -36,10 +36,10 @@ cc_library( srcs = ["arena_planner.cc"], hdrs = ["arena_planner.h"], deps = [ - ":context", ":graph_info", ":memory_planner", ":simple_memory_arena", + "//tensorflow/contrib/lite/c:c_api_internal", ], ) @@ -54,6 +54,7 @@ cc_test( deps = [ ":arena_planner", "//tensorflow/contrib/lite/testing:util", + "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_googletest//:gtest", ], @@ -63,27 +64,27 @@ cc_test( # TODO(aselle): Resolve problems preventing C99 usage. cc_library( name = "context", - srcs = ["context.c"], hdrs = ["context.h"], + deps = ["//tensorflow/contrib/lite/c:c_api_internal"], ) cc_library( name = "graph_info", hdrs = ["graph_info.h"], - deps = [":context"], + deps = ["//tensorflow/contrib/lite/c:c_api_internal"], ) cc_library( name = "memory_planner", hdrs = ["memory_planner.h"], - deps = [":context"], + deps = ["//tensorflow/contrib/lite/c:c_api_internal"], ) cc_library( name = "simple_memory_arena", srcs = ["simple_memory_arena.cc"], hdrs = ["simple_memory_arena.h"], - deps = [":context"], + deps = ["//tensorflow/contrib/lite/c:c_api_internal"], ) cc_library( @@ -91,7 +92,7 @@ cc_library( hdrs = [ "builtin_op_data.h", ], - deps = [":context"], + deps = ["//tensorflow/contrib/lite/c:c_api_internal"], ) cc_library( @@ -121,12 +122,12 @@ cc_library( name = "framework", srcs = [ "allocation.cc", - "error_reporter.cc", "graph_info.cc", "interpreter.cc", "model.cc", - "op_resolver.cc", + "mutable_op_resolver.cc", "optional_debug_tools.cc", + "stderr_reporter.cc", ] + select({ "//tensorflow:android": [ "nnapi_delegate.cc", @@ -149,9 +150,11 @@ cc_library( "graph_info.h", "interpreter.h", "model.h", + "mutable_op_resolver.h", "nnapi_delegate.h", "op_resolver.h", "optional_debug_tools.h", + "stderr_reporter.h", ], copts = tflite_copts(), linkopts = [ @@ -164,14 +167,14 @@ cc_library( }), deps = [ ":arena_planner", - ":builtin_op_data", - ":context", ":graph_info", ":memory_planner", ":schema_fbs_version", ":simple_memory_arena", ":string", ":util", + "//tensorflow/contrib/lite/c:c_api_internal", + "//tensorflow/contrib/lite/core/api", "//tensorflow/contrib/lite/kernels:eigen_support", "//tensorflow/contrib/lite/kernels:gemm_support", "//tensorflow/contrib/lite/nnapi:nnapi_lib", @@ -210,6 +213,8 @@ cc_test( deps = [ ":framework", ":string_util", + "//tensorflow/contrib/lite/c:c_api_internal", + "//tensorflow/contrib/lite/core/api", "//tensorflow/contrib/lite/kernels:builtin_ops", "//tensorflow/contrib/lite/kernels:kernel_util", "//tensorflow/contrib/lite/kernels/internal:tensor_utils", @@ -259,6 +264,8 @@ cc_test( ], deps = [ ":framework", + "//tensorflow/contrib/lite/c:c_api_internal", + "//tensorflow/contrib/lite/core/api", "//tensorflow/contrib/lite/testing:util", "@com_google_googletest//:gtest", ], @@ -266,9 +273,9 @@ cc_test( # Test OpResolver. cc_test( - name = "op_resolver_test", + name = "mutable_op_resolver_test", size = "small", - srcs = ["op_resolver_test.cc"], + srcs = ["mutable_op_resolver_test.cc"], tags = ["no_oss"], deps = [ ":framework", @@ -277,24 +284,12 @@ cc_test( ], ) -# Test the C extension API code. -cc_test( - name = "context_test", - size = "small", - srcs = ["context_test.cc"], - deps = [ - ":framework", - "//tensorflow/contrib/lite/testing:util", - "@com_google_googletest//:gtest", - ], -) - cc_library( name = "util", srcs = ["util.cc"], hdrs = ["util.h"], deps = [ - ":context", + "//tensorflow/contrib/lite/c:c_api_internal", ], ) @@ -304,7 +299,6 @@ cc_test( srcs = ["util_test.cc"], tags = ["no_oss"], deps = [ - ":context", ":util", "//tensorflow/contrib/lite/testing:util", "@com_google_googletest//:gtest", |