From b3fe4d93e6e4b76e1a9da3677890f96f29897dad Mon Sep 17 00:00:00 2001 From: Laura Pak Date: Tue, 27 Aug 2024 14:59:38 -0700 Subject: [PATCH] Remove tf/lite/kernels/shim:tf_headers from tf/core:framework PiperOrigin-RevId: 668152477 --- tensorflow_text/core/kernels/BUILD | 2 ++ tensorflow_text/core/ops/BUILD | 11 ++++++----- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/tensorflow_text/core/kernels/BUILD b/tensorflow_text/core/kernels/BUILD index 11c3bacb9..c20f5c9db 100644 --- a/tensorflow_text/core/kernels/BUILD +++ b/tensorflow_text/core/kernels/BUILD @@ -796,6 +796,7 @@ tf_cc_library( "@com_google_absl//absl/types:span", # lite/kernels/shim:op_kernel tensorflow dep, # lite/kernels/shim:shape tensorflow dep, + # lite/kernels/shim:status_macros tensorflow dep, # lite/kernels/shim:tensor_view tensorflow dep, ], ) @@ -1185,6 +1186,7 @@ tf_cc_library( "@com_google_absl//absl/status:statusor", # lite/kernels/shim:op_kernel tensorflow dep, # lite/kernels/shim:shape tensorflow dep, + # lite/kernels/shim:status_macros tensorflow dep, # lite/kernels/shim:tensor_view tensorflow dep, ], ) diff --git a/tensorflow_text/core/ops/BUILD b/tensorflow_text/core/ops/BUILD index 668ddd8b0..f5d08c6a5 100644 --- a/tensorflow_text/core/ops/BUILD +++ b/tensorflow_text/core/ops/BUILD @@ -14,10 +14,10 @@ cc_test( "byte_splitter_op_test.cc", ], deps = [ - # tf:framework tensorflow dep, # tf:test tensorflow dep, # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:byte_splitter_kernel", ], ) @@ -30,10 +30,10 @@ cc_test( "boise_offset_converter_op_test.cc", ], deps = [ - # tf:framework tensorflow dep, # tf:test tensorflow dep, # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:boise_offset_converter_kernel", ], ) @@ -46,10 +46,10 @@ cc_test( "fast_wordpiece_tokenizer_op_test.cc", ], deps = [ - # tf:framework tensorflow dep, # tf:test tensorflow dep, # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:fast_wordpiece_tokenizer_kernel", ], ) @@ -62,10 +62,10 @@ cc_test( "sentence_fragmenter_v2_op_test.cc", ], deps = [ - # tf:framework tensorflow dep, # tf:test tensorflow dep, # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:sentence_fragmenter_v2_kernel", ], ) @@ -83,6 +83,7 @@ cc_test( # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, # tf/framework:tensor_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:round_robin_trimmer_kernel", ], ) @@ -95,11 +96,11 @@ cc_test( "utf8_binarize_op_test.cc", ], deps = [ - # tf:framework tensorflow dep, # tf:protos_all_cc tensorflow dep, # tf:test tensorflow dep, # tf:test_main tensorflow dep, # tf/framework:shape_inference_testutil tensorflow dep, + # lite/kernels/shim:tf_op_shim tensorflow dep, "//tensorflow_text/core/kernels:utf8_binarize_kernel", ], )