diff --git a/tensorflow/lite/experimental/litert/c/BUILD b/tensorflow/lite/experimental/litert/c/BUILD index e111b9ffad0688..3505780bec23dd 100644 --- a/tensorflow/lite/experimental/litert/c/BUILD +++ b/tensorflow/lite/experimental/litert/c/BUILD @@ -56,9 +56,9 @@ cc_library( ":litert_common", ":litert_op_code", "//tensorflow/lite/core/c:c_api_types", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_support", "//tensorflow/lite/experimental/litert/core/model", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/strings:string_view", ], @@ -72,8 +72,8 @@ cc_test( ":litert_model", ":litert_op_code", ":litert_support", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/core/model", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/test:common", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/types:span", diff --git a/tensorflow/lite/experimental/litert/c/litert_model.cc b/tensorflow/lite/experimental/litert/c/litert_model.cc index 471ab6a8ecaec5..74b1fe30a0370c 100644 --- a/tensorflow/lite/experimental/litert/c/litert_model.cc +++ b/tensorflow/lite/experimental/litert/c/litert_model.cc @@ -20,9 +20,9 @@ #include "absl/strings/string_view.h" #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/schema/schema_generated.h" // diff --git a/tensorflow/lite/experimental/litert/c/litert_model_test.cc b/tensorflow/lite/experimental/litert/c/litert_model_test.cc index 87c7f121387241..4f9ad8731794ba 100644 --- a/tensorflow/lite/experimental/litert/c/litert_model_test.cc +++ b/tensorflow/lite/experimental/litert/c/litert_model_test.cc @@ -25,8 +25,8 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" #include "tensorflow/lite/experimental/litert/c/litert_support.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/test/common.h" #include "tensorflow/lite/schema/schema_generated.h" diff --git a/tensorflow/lite/experimental/litert/cc/BUILD b/tensorflow/lite/experimental/litert/cc/BUILD index dbf53b8cd9385f..847044bc5b0344 100644 --- a/tensorflow/lite/experimental/litert/cc/BUILD +++ b/tensorflow/lite/experimental/litert/cc/BUILD @@ -159,3 +159,27 @@ cc_test( "@com_google_googletest//:gtest_main", ], ) + +cc_library( + name = "litert_buffer_ref", + hdrs = [ + "litert_buffer_ref.h", + ], + deps = [ + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + ], +) + +cc_test( + name = "litert_buffer_ref_test", + srcs = ["litert_buffer_ref_test.cc"], + deps = [ + ":litert_buffer_ref", + "//tensorflow/lite/experimental/litert/core/util:flatbuffer_tools", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", + ], +) diff --git a/tensorflow/lite/experimental/litert/core/util/buffer_ref.h b/tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h similarity index 98% rename from tensorflow/lite/experimental/litert/core/util/buffer_ref.h rename to tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h index 9a921c112a23cf..04592c3d2f97ad 100644 --- a/tensorflow/lite/experimental/litert/core/util/buffer_ref.h +++ b/tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h @@ -12,15 +12,14 @@ // See the License for the specific language governing permissions and // limitations under the License. -#ifndef TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CORE_UTIL_BUFFER_REF_H_ -#define TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CORE_UTIL_BUFFER_REF_H_ +#ifndef TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CC_LITERT_BUFFER_REF_H_ +#define TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CC_LITERT_BUFFER_REF_H_ #include #include #include #include #include -#include #include #include #include @@ -348,4 +347,4 @@ OwningBufferRef(const char*) -> OwningBufferRef; } // namespace litert -#endif // TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CORE_UTIL_BUFFER_REF_H_ +#endif // TENSORFLOW_LITE_EXPERIMENTAL_LITERT_CC_LITERT_BUFFER_REF_H_ diff --git a/tensorflow/lite/experimental/litert/core/util/buffer_ref_test.cc b/tensorflow/lite/experimental/litert/cc/litert_buffer_ref_test.cc similarity index 99% rename from tensorflow/lite/experimental/litert/core/util/buffer_ref_test.cc rename to tensorflow/lite/experimental/litert/cc/litert_buffer_ref_test.cc index c0561c31b5db09..adb0b622435cc2 100644 --- a/tensorflow/lite/experimental/litert/core/util/buffer_ref_test.cc +++ b/tensorflow/lite/experimental/litert/cc/litert_buffer_ref_test.cc @@ -12,7 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include #include diff --git a/tensorflow/lite/experimental/litert/core/BUILD b/tensorflow/lite/experimental/litert/core/BUILD index 37d9ed4eee61a5..4821adc5d21d20 100644 --- a/tensorflow/lite/experimental/litert/core/BUILD +++ b/tensorflow/lite/experimental/litert/core/BUILD @@ -24,8 +24,8 @@ cc_library( deps = [ "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_logging", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/strings:string_view", @@ -41,8 +41,8 @@ cc_test( deps = [ ":byte_code_util", "//tensorflow/lite/experimental/litert/c:litert_common", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/test:common", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", @@ -61,8 +61,8 @@ cc_library( "//tensorflow/lite/experimental/litert/c:litert_logging", "//tensorflow/lite/experimental/litert/c:litert_model", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@llvm-project//llvm:Support", diff --git a/tensorflow/lite/experimental/litert/core/byte_code_util.cc b/tensorflow/lite/experimental/litert/core/byte_code_util.cc index 7adfe2a1085d0c..fb5170154fddbc 100644 --- a/tensorflow/lite/experimental/litert/core/byte_code_util.cc +++ b/tensorflow/lite/experimental/litert/core/byte_code_util.cc @@ -24,8 +24,8 @@ #include "absl/strings/string_view.h" #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_logging.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" namespace litert::internal { diff --git a/tensorflow/lite/experimental/litert/core/byte_code_util.h b/tensorflow/lite/experimental/litert/core/byte_code_util.h index 7e17c9bdab7def..36aa590d4d97f8 100644 --- a/tensorflow/lite/experimental/litert/core/byte_code_util.h +++ b/tensorflow/lite/experimental/litert/core/byte_code_util.h @@ -20,8 +20,8 @@ #include #include "tensorflow/lite/experimental/litert/c/litert_common.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" namespace litert::internal { diff --git a/tensorflow/lite/experimental/litert/core/byte_code_util_test.cc b/tensorflow/lite/experimental/litert/core/byte_code_util_test.cc index e4c0fb680f505b..fd8789a84edce8 100644 --- a/tensorflow/lite/experimental/litert/core/byte_code_util_test.cc +++ b/tensorflow/lite/experimental/litert/core/byte_code_util_test.cc @@ -24,8 +24,8 @@ #include #include "absl/strings/string_view.h" #include "tensorflow/lite/experimental/litert/c/litert_common.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/test/common.h" namespace litert::internal { diff --git a/tensorflow/lite/experimental/litert/core/graph_tools.h b/tensorflow/lite/experimental/litert/core/graph_tools.h index 266b3df6700886..707a4a376523f6 100644 --- a/tensorflow/lite/experimental/litert/core/graph_tools.h +++ b/tensorflow/lite/experimental/litert/core/graph_tools.h @@ -28,8 +28,8 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #define MATCH_TRUE(v) \ { \ diff --git a/tensorflow/lite/experimental/litert/core/model/BUILD b/tensorflow/lite/experimental/litert/core/model/BUILD index 42e0d217f1a7f7..f3ad53efb7bc7e 100644 --- a/tensorflow/lite/experimental/litert/core/model/BUILD +++ b/tensorflow/lite/experimental/litert/core/model/BUILD @@ -32,8 +32,8 @@ cc_library( "//tensorflow/lite/core/c:c_api_types", "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/strings:string_view", ], @@ -44,7 +44,7 @@ cc_test( srcs = ["model_test.cc"], deps = [ ":model", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/test:common", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/strings:string_view", @@ -65,9 +65,9 @@ cc_library( "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_logging", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_model", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/core/util:flatbuffer_tools", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/strings:string_view", @@ -84,10 +84,10 @@ cc_test( ":model_serialize", "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_model", "//tensorflow/lite/experimental/litert/cc:litert_support", "//tensorflow/lite/experimental/litert/core:graph_tools", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/core/util:flatbuffer_tools", "//tensorflow/lite/experimental/litert/test:common", "@com_google_absl//absl/types:span", @@ -104,9 +104,9 @@ cc_library( ":model_util", "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_model", "//tensorflow/lite/experimental/litert/cc:litert_support", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/core/util:flatbuffer_tools", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/container:flat_hash_map", diff --git a/tensorflow/lite/experimental/litert/core/model/model.cc b/tensorflow/lite/experimental/litert/core/model/model.cc index e4d8cb4915f3f8..823f6b9e5111a5 100644 --- a/tensorflow/lite/experimental/litert/core/model/model.cc +++ b/tensorflow/lite/experimental/litert/core/model/model.cc @@ -22,8 +22,8 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/schema/schema_generated.h" using litert::BufferRef; diff --git a/tensorflow/lite/experimental/litert/core/model/model.h b/tensorflow/lite/experimental/litert/core/model/model.h index ef4a61be6424f7..8a264a0fd22fe0 100644 --- a/tensorflow/lite/experimental/litert/core/model/model.h +++ b/tensorflow/lite/experimental/litert/core/model/model.h @@ -23,8 +23,8 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/schema/schema_generated.h" // diff --git a/tensorflow/lite/experimental/litert/core/model/model_file_test.cc b/tensorflow/lite/experimental/litert/core/model/model_file_test.cc index 47a3e1638a4302..23c593d0c31b86 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_file_test.cc +++ b/tensorflow/lite/experimental/litert/core/model/model_file_test.cc @@ -26,13 +26,13 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" #include "tensorflow/lite/experimental/litert/core/graph_tools.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" #include "tensorflow/lite/experimental/litert/core/model/model_load.h" #include "tensorflow/lite/experimental/litert/core/model/model_serialize.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/util/flatbuffer_tools.h" #include "tensorflow/lite/experimental/litert/test/common.h" diff --git a/tensorflow/lite/experimental/litert/core/model/model_load.cc b/tensorflow/lite/experimental/litert/core/model/model_load.cc index d8c9975703e22b..4ab56cdf45562f 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_load.cc +++ b/tensorflow/lite/experimental/litert/core/model/model_load.cc @@ -27,11 +27,11 @@ #include "tensorflow/lite/experimental/litert/c/litert_logging.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" #include "tensorflow/lite/experimental/litert/core/model/model_util.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/util/flatbuffer_tools.h" #include "tensorflow/lite/schema/schema_generated.h" #include "tensorflow/lite/stderr_reporter.h" diff --git a/tensorflow/lite/experimental/litert/core/model/model_load.h b/tensorflow/lite/experimental/litert/core/model/model_load.h index 8c019cfecaaff5..44c4450085dbdb 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_load.h +++ b/tensorflow/lite/experimental/litert/core/model/model_load.h @@ -32,9 +32,9 @@ LiteRtStatus LiteRtLoadModelFromMemory(const uint8_t* buf, size_t buf_size, #ifdef __cplusplus } +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" namespace litert::internal { diff --git a/tensorflow/lite/experimental/litert/core/model/model_serialize.cc b/tensorflow/lite/experimental/litert/core/model/model_serialize.cc index 67202c4128cbc5..f553ed97b8c1a3 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_serialize.cc +++ b/tensorflow/lite/experimental/litert/core/model/model_serialize.cc @@ -27,11 +27,11 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" #include "tensorflow/lite/experimental/litert/core/model/model_util.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/util/flatbuffer_tools.h" #include "tensorflow/lite/schema/schema_generated.h" diff --git a/tensorflow/lite/experimental/litert/core/model/model_serialize.h b/tensorflow/lite/experimental/litert/core/model/model_serialize.h index 6b08d15c5c035d..078ad3569932ec 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_serialize.h +++ b/tensorflow/lite/experimental/litert/core/model/model_serialize.h @@ -30,9 +30,9 @@ LiteRtStatus LiteRtSerializeModel(LiteRtModel model, uint8_t** buf, #ifdef __cplusplus } +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" namespace litert::internal { diff --git a/tensorflow/lite/experimental/litert/core/model/model_test.cc b/tensorflow/lite/experimental/litert/core/model/model_test.cc index dd1dd2bbe7e533..b00d079365c315 100644 --- a/tensorflow/lite/experimental/litert/core/model/model_test.cc +++ b/tensorflow/lite/experimental/litert/core/model/model_test.cc @@ -19,7 +19,7 @@ #include #include "absl/strings/string_view.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/test/common.h" #include "tensorflow/lite/schema/schema_generated.h" diff --git a/tensorflow/lite/experimental/litert/core/util/BUILD b/tensorflow/lite/experimental/litert/core/util/BUILD index 3fba4c4828951d..2045578f42bb64 100644 --- a/tensorflow/lite/experimental/litert/core/util/BUILD +++ b/tensorflow/lite/experimental/litert/core/util/BUILD @@ -17,30 +17,6 @@ package( default_visibility = ["//tensorflow/lite/experimental/litert:__subpackages__"], ) -cc_library( - name = "buffer_ref", - hdrs = [ - "buffer_ref.h", - ], - deps = [ - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - ], -) - -cc_test( - name = "buffer_ref_test", - srcs = ["buffer_ref_test.cc"], - deps = [ - ":buffer_ref", - ":flatbuffer_tools", - "@com_google_absl//absl/strings:string_view", - "@com_google_absl//absl/types:span", - "@com_google_googletest//:gtest_main", - ], -) - cc_library( name = "flatbuffer_tools", srcs = ["flatbuffer_tools.cc"], @@ -48,7 +24,7 @@ cc_library( "flatbuffer_tools.h", ], deps = [ - ":buffer_ref", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/schema:schema_fbs", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", diff --git a/tensorflow/lite/experimental/litert/tools/BUILD b/tensorflow/lite/experimental/litert/tools/BUILD index 17b17431f68eda..888e97850cfe57 100644 --- a/tensorflow/lite/experimental/litert/tools/BUILD +++ b/tensorflow/lite/experimental/litert/tools/BUILD @@ -30,6 +30,7 @@ cc_library( "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_model", "//tensorflow/lite/experimental/litert/c:litert_support", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/cc:litert_model", "//tensorflow/lite/experimental/litert/cc:litert_support", "//tensorflow/lite/experimental/litert/compiler/plugin:algo", @@ -37,7 +38,6 @@ cc_library( "//tensorflow/lite/experimental/litert/core:byte_code_util", "//tensorflow/lite/experimental/litert/core/model:model_load", "//tensorflow/lite/experimental/litert/core/model:model_serialize", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/core/util:flatbuffer_tools", "@com_google_absl//absl/log:absl_check", "@com_google_absl//absl/strings:str_format", @@ -63,11 +63,11 @@ cc_test( "//tensorflow/lite/experimental/litert/c:litert_common", "//tensorflow/lite/experimental/litert/c:litert_model", "//tensorflow/lite/experimental/litert/c:litert_op_code", + "//tensorflow/lite/experimental/litert/cc:litert_buffer_ref", "//tensorflow/lite/experimental/litert/core:byte_code_util", "//tensorflow/lite/experimental/litert/core:graph_tools", "//tensorflow/lite/experimental/litert/core/model", "//tensorflow/lite/experimental/litert/core/model:model_load", - "//tensorflow/lite/experimental/litert/core/util:buffer_ref", "//tensorflow/lite/experimental/litert/test:common", "@com_google_absl//absl/log:absl_check", "@com_google_absl//absl/strings:string_view", diff --git a/tensorflow/lite/experimental/litert/tools/apply_plugin.cc b/tensorflow/lite/experimental/litert/tools/apply_plugin.cc index 6239b994f43cf2..e87244c019773f 100644 --- a/tensorflow/lite/experimental/litert/tools/apply_plugin.cc +++ b/tensorflow/lite/experimental/litert/tools/apply_plugin.cc @@ -31,6 +31,7 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_support.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/cc/litert_model.h" #include "tensorflow/lite/experimental/litert/cc/litert_support.h" #include "tensorflow/lite/experimental/litert/compiler/plugin/algo.h" @@ -38,7 +39,6 @@ #include "tensorflow/lite/experimental/litert/core/byte_code_util.h" #include "tensorflow/lite/experimental/litert/core/model/model_load.h" #include "tensorflow/lite/experimental/litert/core/model/model_serialize.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/util/flatbuffer_tools.h" #include "tensorflow/lite/experimental/litert/tools/dump.h" #include "tensorflow/lite/experimental/litert/tools/tool_display.h" diff --git a/tensorflow/lite/experimental/litert/tools/apply_plugin_test.cc b/tensorflow/lite/experimental/litert/tools/apply_plugin_test.cc index fd6d2e9c904fdb..27d54b68eec359 100644 --- a/tensorflow/lite/experimental/litert/tools/apply_plugin_test.cc +++ b/tensorflow/lite/experimental/litert/tools/apply_plugin_test.cc @@ -26,11 +26,11 @@ #include "tensorflow/lite/experimental/litert/c/litert_common.h" #include "tensorflow/lite/experimental/litert/c/litert_model.h" #include "tensorflow/lite/experimental/litert/c/litert_op_code.h" +#include "tensorflow/lite/experimental/litert/cc/litert_buffer_ref.h" #include "tensorflow/lite/experimental/litert/core/byte_code_util.h" #include "tensorflow/lite/experimental/litert/core/graph_tools.h" #include "tensorflow/lite/experimental/litert/core/model/model.h" #include "tensorflow/lite/experimental/litert/core/model/model_load.h" -#include "tensorflow/lite/experimental/litert/core/util/buffer_ref.h" #include "tensorflow/lite/experimental/litert/test/common.h" namespace litert::tools {