Skip to content

Commit 7d067cc

Browse files
committed
Update BUILD file
1 parent d1b94e2 commit 7d067cc

File tree

1 file changed

+11
-10
lines changed
  • larq_compute_engine/mlir

1 file changed

+11
-10
lines changed

larq_compute_engine/mlir/BUILD

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,10 @@ td_library(
1111
srcs = [
1212
"ir/lce_ops.td",
1313
],
14-
includes = ["include"],
15-
deps = [],
14+
deps = [
15+
"@org_tensorflow//tensorflow/compiler/mlir/lite/quantization:quantization_td_files",
16+
"@llvm-project//mlir:SideEffectTdFiles",
17+
],
1618
)
1719

1820
gentbl(
@@ -27,8 +29,7 @@ gentbl(
2729
td_file = "ir/lce_ops.td",
2830
td_includes = ["external/org_tensorflow"],
2931
td_srcs = [
30-
"@org_tensorflow//tensorflow/compiler/mlir/lite/quantization:quantization_td_files",
31-
"@llvm-project//mlir:SideEffectTdFiles",
32+
":lce_ops_td_file"
3233
],
3334
)
3435

@@ -55,7 +56,7 @@ gentbl(
5556
td_file = "transforms/prepare_patterns_target_arm.td",
5657
td_includes = ["external/org_tensorflow"],
5758
td_srcs = [
58-
"ir/lce_ops.td",
59+
":lce_ops_td_file",
5960
"transforms/op_removal_patterns.td",
6061
"transforms/prepare_patterns_common.td",
6162
"@llvm-project//mlir:StdOpsTdFiles",
@@ -73,7 +74,7 @@ gentbl(
7374
td_file = "transforms/prepare_patterns_common.td",
7475
td_includes = ["external/org_tensorflow"],
7576
td_srcs = [
76-
"ir/lce_ops.td",
77+
":lce_ops_td_file",
7778
"transforms/op_removal_patterns.td",
7879
"@llvm-project//mlir:StdOpsTdFiles",
7980
"@org_tensorflow//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
@@ -90,7 +91,7 @@ gentbl(
9091
td_file = "transforms/optimize_patterns_target_arm.td",
9192
td_includes = ["external/org_tensorflow"],
9293
td_srcs = [
93-
"ir/lce_ops.td",
94+
":lce_ops_td_file",
9495
"transforms/optimize_patterns_common.td",
9596
"@org_tensorflow//tensorflow/compiler/mlir/lite:tensorflow_lite_ops_td_files",
9697
"@llvm-project//mlir:StdOpsTdFiles",
@@ -106,7 +107,7 @@ gentbl(
106107
td_file = "transforms/optimize_patterns_common.td",
107108
td_includes = ["external/org_tensorflow"],
108109
td_srcs = [
109-
"ir/lce_ops.td",
110+
":lce_ops_td_file",
110111
"@org_tensorflow//tensorflow/compiler/mlir/lite:tensorflow_lite_ops_td_files",
111112
"@llvm-project//mlir:StdOpsTdFiles",
112113
],
@@ -121,7 +122,7 @@ gentbl(
121122
td_file = "transforms/bitpack_weights_patterns.td",
122123
td_includes = ["external/org_tensorflow"],
123124
td_srcs = [
124-
"ir/lce_ops.td",
125+
":lce_ops_td_file",
125126
"transforms/op_removal_patterns.td",
126127
"@org_tensorflow//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
127128
"@org_tensorflow//tensorflow/compiler/mlir/lite:tensorflow_lite_ops_td_files",
@@ -138,7 +139,7 @@ gentbl(
138139
td_file = "transforms/quantize_patterns.td",
139140
td_includes = ["external/org_tensorflow"],
140141
td_srcs = [
141-
"ir/lce_ops.td",
142+
":lce_ops_td_file",
142143
"@org_tensorflow//tensorflow/compiler/mlir/lite:tensorflow_lite_ops_td_files",
143144
],
144145
)

0 commit comments

Comments
 (0)