diff options
| author |  Pete Warden <petewarden@google.com> | 2018-09-07 17:36:59 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 17:40:10 -0700 |
| commit | 9982fd6c8831cbd2f58954f79ea71f26660393bc (patch) | |
| tree | 108907bde953d0d70ee5d3b8323a99bb9b681563 /tensorflow/contrib/lite/experimental | |
| parent | edda5e39e4e93ba60e4d31b6ecb1c295dead29c8 (diff) | |
Modularize TF Lite interface definitions and reorganize file structure
PiperOrigin-RevId: 212064501
Diffstat (limited to 'tensorflow/contrib/lite/experimental')
3 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/experimental/c/BUILD b/tensorflow/contrib/lite/experimental/c/BUILD index 8fc07e8eb7..ea4a543252 100644 --- a/tensorflow/contrib/lite/experimental/c/BUILD +++ b/tensorflow/contrib/lite/experimental/c/BUILD @@ -78,6 +78,7 @@ cc_test( data = ["//tensorflow/contrib/lite:testdata/add.bin"], deps = [ ":c_api", + "//tensorflow/contrib/lite:context", "//tensorflow/contrib/lite:kernel_api", "//tensorflow/contrib/lite/testing:util", "@com_google_googletest//:gtest", diff --git a/tensorflow/contrib/lite/experimental/kernels/BUILD b/tensorflow/contrib/lite/experimental/kernels/BUILD index 9c06c4ebd9..4786cc62f9 100644 --- a/tensorflow/contrib/lite/experimental/kernels/BUILD +++ b/tensorflow/contrib/lite/experimental/kernels/BUILD @@ -53,6 +53,7 @@ cc_library( "//tensorflow/contrib/lite:builtin_op_data", "//tensorflow/contrib/lite:framework", "//tensorflow/contrib/lite:string_util", + "//tensorflow/contrib/lite/c:c_api_internal", "//tensorflow/contrib/lite/kernels:builtin_ops", "//tensorflow/contrib/lite/kernels:gemm_support", "//tensorflow/contrib/lite/kernels:kernel_util", @@ -61,8 +62,8 @@ cc_library( "//tensorflow/contrib/lite/kernels/internal:optimized", "//tensorflow/contrib/lite/kernels/internal:optimized_base", "//tensorflow/contrib/lite/kernels/internal:quantization_util", - "//tensorflow/contrib/lite/kernels/internal:reference", "//tensorflow/contrib/lite/kernels/internal:reference_base", + "//tensorflow/contrib/lite/kernels/internal:tensor", "//tensorflow/contrib/lite/kernels/internal:tensor_utils", "@flatbuffers", ], diff --git a/tensorflow/contrib/lite/experimental/kernels/ctc_beam_search_decoder.cc b/tensorflow/contrib/lite/experimental/kernels/ctc_beam_search_decoder.cc index 121997dcb2..8442c4d46c 100644 --- a/tensorflow/contrib/lite/experimental/kernels/ctc_beam_search_decoder.cc +++ b/tensorflow/contrib/lite/experimental/kernels/ctc_beam_search_decoder.cc @@ -14,7 +14,7 @@ limitations under the License. ==============================================================================*/ #include <vector> #include "flatbuffers/flexbuffers.h" // flatbuffers -#include "tensorflow/contrib/lite/context.h" +#include "tensorflow/contrib/lite/c/c_api_internal.h" #include "tensorflow/contrib/lite/experimental/kernels/ctc_beam_search.h" #include "tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h" #include "tensorflow/contrib/lite/kernels/internal/tensor.h" |