Skip to content

Commit 7e02e4b

Browse files
yijie-yangtensorflower-gardener
authored andcommitted
Moved some functions from tensorflow/compiler/mlir/lite/flatbuffer_import.cc to tensorflow/compiler/mlir/lite/utils/const_tensor_utils.h for shared usage
PiperOrigin-RevId: 599023637
1 parent c8ff9d7 commit 7e02e4b

File tree

4 files changed

+585
-434
lines changed

4 files changed

+585
-434
lines changed

tensorflow/compiler/mlir/lite/BUILD

Lines changed: 29 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1114,9 +1114,9 @@ cc_library(
11141114
"flatbuffer_import.h",
11151115
],
11161116
deps = [
1117+
":const_tensor_utils",
11171118
":convert_type",
11181119
":flatbuffer_tflite_operator_lib",
1119-
":low_bit_utils",
11201120
":offset_buffer",
11211121
":size_utils",
11221122
":tensorflow_lite",
@@ -1132,16 +1132,14 @@ cc_library(
11321132
"//tensorflow/core/platform:errors",
11331133
"//tensorflow/core/platform:status",
11341134
"//tensorflow/lite:framework",
1135-
"//tensorflow/lite:string_util",
11361135
"//tensorflow/lite/experimental/remat:metadata_util",
11371136
"//tensorflow/lite/schema:schema_fbs_with_mutable",
11381137
"//tensorflow/lite/schema:schema_utils",
1139-
"@com_google_absl//absl/base",
11401138
"@com_google_absl//absl/container:flat_hash_map",
11411139
"@com_google_absl//absl/container:flat_hash_set",
11421140
"@com_google_absl//absl/status",
11431141
"@com_google_absl//absl/strings",
1144-
"@eigen_archive//:eigen3",
1142+
"@llvm-project//llvm:Analysis",
11451143
"@llvm-project//llvm:Support",
11461144
"@llvm-project//mlir:ArithDialect",
11471145
"@llvm-project//mlir:FuncDialect",
@@ -1150,7 +1148,8 @@ cc_library(
11501148
"@llvm-project//mlir:QuantOps",
11511149
"@llvm-project//mlir:Support",
11521150
"@llvm-project//mlir:TranslateLib",
1153-
"@local_xla//xla:statusor",
1151+
"@local_tsl//tsl/platform:status",
1152+
"@local_tsl//tsl/platform:statusor",
11541153
"@stablehlo//:stablehlo_ops",
11551154
],
11561155
)
@@ -1442,6 +1441,31 @@ cc_library(
14421441
hdrs = ["offset_buffer.h"],
14431442
)
14441443

1444+
cc_library(
1445+
name = "const_tensor_utils",
1446+
srcs = ["utils/const_tensor_utils.cc"],
1447+
hdrs = ["utils/const_tensor_utils.h"],
1448+
deps = [
1449+
":convert_type",
1450+
":low_bit_utils",
1451+
"//tensorflow/compiler/mlir/tensorflow:dynamic_shape_utils",
1452+
"//tensorflow/compiler/mlir/tensorflow:tensorflow_types",
1453+
"//tensorflow/core:protos_all_cc",
1454+
"//tensorflow/lite:string_util",
1455+
"//tensorflow/lite/schema:schema_fbs",
1456+
"@com_google_absl//absl/base",
1457+
"@com_google_absl//absl/status",
1458+
"@com_google_absl//absl/status:statusor",
1459+
"@com_google_absl//absl/strings",
1460+
"@eigen_archive//:eigen3",
1461+
"@llvm-project//llvm:Support",
1462+
"@llvm-project//mlir:IR",
1463+
"@llvm-project//mlir:QuantOps",
1464+
"@llvm-project//mlir:Support",
1465+
"@local_tsl//tsl/platform:statusor",
1466+
],
1467+
)
1468+
14451469
tf_cc_test(
14461470
name = "offset_buffer_test",
14471471
srcs = ["offset_buffer_test.cc"],

0 commit comments

Comments
 (0)