Creates framework_lib target.

PiperOrigin-RevId: 301656623
Change-Id: Ied2158a67bfb0fa29753d201610d2967f57d3504
This commit is contained in:
Nupur Garg 2020-03-18 13:36:10 -07:00 committed by TensorFlower Gardener
parent ef46437ea4
commit b2a5472997
2 changed files with 52 additions and 18 deletions

View File

@ -62,6 +62,22 @@ TFLITE_DEFAULT_COPTS = if_not_windows([
"-Wno-extern-c-compat",
])
FRAMEWORK_LIB_HDRS = [
"allocation.h",
"context.h",
"context_util.h",
"core/macros.h",
"core/subgraph.h",
"error_reporter.h",
"graph_info.h",
"interpreter.h",
"model.h",
"mutable_op_resolver.h",
"op_resolver.h",
"optional_debug_tools.h",
"stderr_reporter.h",
]
cc_library(
name = "version",
hdrs = ["version.h"],
@ -200,9 +216,8 @@ cc_library(
],
)
# TODO(ahentz): investigate dependency on gemm_support requiring usage of tf_copts.
cc_library(
name = "framework",
name = "framework_lib",
srcs = [
"core/subgraph.cc",
"graph_info.cc",
@ -212,23 +227,42 @@ cc_library(
"optional_debug_tools.cc",
"stderr_reporter.cc",
],
hdrs = [
"allocation.h",
"context.h",
"context_util.h",
"core/macros.h",
"core/subgraph.h",
"error_reporter.h",
"graph_info.h",
"interpreter.h",
"model.h",
"mutable_op_resolver.h",
"op_resolver.h",
"optional_debug_tools.h",
"stderr_reporter.h",
hdrs = FRAMEWORK_LIB_HDRS,
copts = tflite_copts() + TFLITE_DEFAULT_COPTS,
visibility = [
"//tensorflow/lite:__subpackages__",
],
deps = [
":allocation",
":arena_planner",
":external_cpu_backend_context",
":graph_info",
":memory_planner",
":minimal_logging",
":simple_memory_arena",
":string",
":type_to_tflitetype",
":util",
":version",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core/api",
"//tensorflow/lite/delegates/nnapi:nnapi_delegate",
"//tensorflow/lite/experimental/resource",
"//tensorflow/lite/nnapi:nnapi_implementation",
"//tensorflow/lite/schema:schema_fbs",
],
alwayslink = 1,
)
# TODO(ahentz): investigate dependency on gemm_support requiring usage of tf_copts.
cc_library(
name = "framework",
srcs = [
],
hdrs = FRAMEWORK_LIB_HDRS,
copts = tflite_copts() + TFLITE_DEFAULT_COPTS,
deps = [
":framework_lib",
":allocation",
":arena_planner",
":external_cpu_backend_context",

View File

@ -526,7 +526,7 @@ cc_library(
":lstm_shared",
":op_macros",
":padding",
"//tensorflow/lite:framework",
"//tensorflow/lite:framework_lib",
"//tensorflow/lite:minimal_logging",
"//tensorflow/lite:string_util",
"//tensorflow/lite/c:common",
@ -660,7 +660,7 @@ cc_library(
],
deps = [
":builtin_op_kernels",
"//tensorflow/lite:framework",
"//tensorflow/lite:framework_lib",
"//tensorflow/lite/c:common",
],
)