experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #37368 from lgeiger:cleanup-mlir-tflite-build-target

Browse Source 
PiperOrigin-RevId: 299565200
Change-Id: I913f0a1230ae5ccbe8077c883f9492f81d2b8896
This commit is contained in:
TensorFlower Gardener 2020-03-07 09:36:23 -08:00
commit d8ee1d31d5
2 changed files with 23 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/mlir/lite/BUILD
View File

@ -697,9 +697,9 @@ cc_library(
"//tensorflow/compiler/mlir/lite/quantization:quantization_config", "//tensorflow/compiler/mlir/lite/quantization:quantization_config",
"//tensorflow/compiler/mlir/lite/quantization:quantization_passes", "//tensorflow/compiler/mlir/lite/quantization:quantization_passes",
"//tensorflow/compiler/mlir/tensorflow", "//tensorflow/compiler/mlir/tensorflow",
"//tensorflow/compiler/mlir/tensorflow:decode_constant_pass",
"//tensorflow/compiler/mlir/tensorflow:tensorflow_dialect_registration",
"//tensorflow/compiler/mlir/tensorflow:tensorflow_passes", "//tensorflow/compiler/mlir/tensorflow:tensorflow_passes",
"//tensorflow/compiler/mlir/tensorflow:tf_dialect_lib",
"//tensorflow/compiler/mlir/tensorflow:tf_dialect_passes",
"//tensorflow/compiler/mlir/tensorflow:tf_graph_optimization_pass", "//tensorflow/compiler/mlir/tensorflow:tf_graph_optimization_pass",
"//tensorflow/compiler/mlir/tensorflow:translate_lib", "//tensorflow/compiler/mlir/tensorflow:translate_lib",
"@llvm-project//mlir:Analysis", "@llvm-project//mlir:Analysis",
@ -725,6 +725,7 @@ cc_library(
":tensorflow_lite_quantize", ":tensorflow_lite_quantize",
"//tensorflow/compiler/mlir/lite/quantization:quantization_config", "//tensorflow/compiler/mlir/lite/quantization:quantization_config",
"//tensorflow/compiler/mlir/tensorflow", "//tensorflow/compiler/mlir/tensorflow",
"//tensorflow/compiler/mlir/tensorflow:decode_constant_pass",
"//tensorflow/compiler/mlir/tensorflow:error_util", "//tensorflow/compiler/mlir/tensorflow:error_util",
"//tensorflow/compiler/mlir/tensorflow:tf_dialect_lib", "//tensorflow/compiler/mlir/tensorflow:tf_dialect_lib",
"//tensorflow/compiler/mlir/tensorflow:tf_dialect_passes", "//tensorflow/compiler/mlir/tensorflow:tf_dialect_passes",
@ -734,11 +735,9 @@ cc_library(
"//tensorflow/lite/tools/optimize:quantize_weights", "//tensorflow/lite/tools/optimize:quantize_weights",
"//tensorflow/stream_executor/lib", "//tensorflow/stream_executor/lib",
"@llvm-project//llvm:support", "@llvm-project//llvm:support",
"@llvm-project//mlir:Analysis",
"@llvm-project//mlir:IR", "@llvm-project//mlir:IR",
"@llvm-project//mlir:Parser", "@llvm-project//mlir:Parser",
"@llvm-project//mlir:Pass", "@llvm-project//mlir:Pass",
"@llvm-project//mlir:QuantOps",
"@llvm-project//mlir:Support", "@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms", "@llvm-project//mlir:Transforms",
], ],

24
tensorflow/compiler/mlir/tensorflow/BUILD
View File

@ -727,18 +727,34 @@ cc_library(
) )
cc_library( cc_library(
name = "tf_dialect_passes", name = "decode_constant_pass",
srcs = [ srcs = [
"transforms/constant_fold.cc",
"transforms/decode_constant.cc", "transforms/decode_constant.cc",
"transforms/dialect_hooks.cc",
], ],
hdrs = [ hdrs = [
"transforms/constant_fold.h",
"transforms/decode_constant.h", "transforms/decode_constant.h",
], ],
deps = [ deps = [
":convert_tensor", ":convert_tensor",
":tensorflow",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
],
alwayslink = 1,
)
cc_library(
name = "tf_dialect_passes",
srcs = [
"transforms/constant_fold.cc",
"transforms/dialect_hooks.cc",
],
hdrs = [
"transforms/constant_fold.h",
],
deps = [
":convert_tensor",
":decode_constant_pass",
":eval_util", ":eval_util",
":tensorflow", ":tensorflow",
":tensorflow_types", ":tensorflow_types",