experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adds TFLRT dependency to framework target.

Browse Source 
PiperOrigin-RevId: 301680495
Change-Id: I1d10bb6f66a0718682fc7db007e963426c64785e
This commit is contained in:
Nupur Garg 2020-03-18 15:20:29 -07:00 committed by TensorFlower Gardener
parent 8b30b82190
commit d9d7b0cb78
2 changed files with 9 additions and 4 deletions
tensorflow/lite
BUILD
experimental/tflite_api_dispatcher
BUILD

9
tensorflow/lite/BUILD
View File

@ -1,5 +1,5 @@
load("//tensorflow:tensorflow.bzl", "if_not_windows", "tf_cc_test") load("//tensorflow:tensorflow.bzl", "if_not_windows", "tf_cc_test")
load("//tensorflow/lite:build_def.bzl", "tflite_cc_shared_object", "tflite_copts") load("//tensorflow/lite:build_def.bzl", "if_tflite_experimental_runtime", "tflite_cc_shared_object", "tflite_copts", "tflite_experimental_runtime_linkopts")
load("//tensorflow/lite/micro:build_def.bzl", "cc_library") load("//tensorflow/lite/micro:build_def.bzl", "cc_library")
load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite") load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite")
@ -261,6 +261,11 @@ cc_library(
], ],
hdrs = FRAMEWORK_LIB_HDRS, hdrs = FRAMEWORK_LIB_HDRS,
copts = tflite_copts() + TFLITE_DEFAULT_COPTS, copts = tflite_copts() + TFLITE_DEFAULT_COPTS,
defines = if_tflite_experimental_runtime(
if_eager = ["TFLITE_EXPERIMENTAL_RUNTIME_EAGER"],
if_non_eager = ["TFLITE_EXPERIMENTAL_RUNTIME_NON_EAGER"],
if_none = [],
),
deps = [ deps = [
":framework_lib", ":framework_lib",
":allocation", ":allocation",
@ -285,7 +290,7 @@ cc_library(
"//tensorflow/lite/profiling:platform_profiler", "//tensorflow/lite/profiling:platform_profiler",
], ],
"//conditions:default": [], "//conditions:default": [],
}), }) + tflite_experimental_runtime_linkopts(),
alwayslink = 1, alwayslink = 1,
) )

4
tensorflow/lite/experimental/tflite_api_dispatcher/BUILD
View File

@ -14,7 +14,7 @@ cc_library(
if_none = [], if_none = [],
), ),
deps = [ deps = [
"//tensorflow/lite:framework", "//tensorflow/lite:framework_lib",
] + tflite_experimental_runtime_linkopts(), ] + tflite_experimental_runtime_linkopts(),
) )
@ -23,7 +23,7 @@ cc_library(
hdrs = ["tflite_api_dispatcher.h"], hdrs = ["tflite_api_dispatcher.h"],
deps = [ deps = [
":tflite_api_dispatcher", ":tflite_api_dispatcher",
"//tensorflow/lite:framework", "//tensorflow/lite:framework_lib",
] + tflite_experimental_runtime_linkopts( ] + tflite_experimental_runtime_linkopts(
if_eager = [ if_eager = [
# "//tensorflow/lite/experimental/tf_runtime/opdef:tflrt_opdefs", # "//tensorflow/lite/experimental/tf_runtime/opdef:tflrt_opdefs",