diff options
author | 2018-10-05 01:22:02 -0700 | |
---|---|---|
committer | 2018-10-05 01:29:25 -0700 | |
commit | 3b94d75a9e10ef8ef33760d0ef6aad326e1353ba (patch) | |
tree | 402934b406e63ccd9cff0faec8a83aba6d58abf3 /tensorflow/contrib/lite/kernels/BUILD | |
parent | 57d31aa599c83014397a22bbb8f1a27a33b0ade3 (diff) |
Merge the different LSTM EvalFloat/EvalHybrid calls into a single file.
PiperOrigin-RevId: 215870962
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index 95e387814d..68636fb070 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -234,11 +234,11 @@ cc_library( ":activation_functor", ":eigen_support", ":kernel_util", + ":lstm_eval", ":op_macros", ":padding", "//tensorflow/contrib/lite:framework", "//tensorflow/contrib/lite:string_util", - "//tensorflow/contrib/lite:util", "//tensorflow/contrib/lite/c:c_api_internal", "//tensorflow/contrib/lite/kernels:gemm_support", "//tensorflow/contrib/lite/kernels/internal:audio_utils", @@ -255,6 +255,17 @@ cc_library( ) cc_library( + name = "lstm_eval", + srcs = ["lstm_eval.cc"], + hdrs = ["lstm_eval.h"], + deps = [ + "//tensorflow/contrib/lite/c:c_api_internal", + "//tensorflow/contrib/lite/kernels/internal:kernel_utils", + "//tensorflow/contrib/lite/kernels/internal:tensor_utils", + ], +) + +cc_library( name = "builtin_ops", srcs = ["register.cc"], hdrs = ["register.h"], |