Creates framework_lib
target.
PiperOrigin-RevId: 301656623 Change-Id: Ied2158a67bfb0fa29753d201610d2967f57d3504
This commit is contained in:
parent
ef46437ea4
commit
b2a5472997
@ -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",
|
||||
|
@ -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",
|
||||
],
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user