File tree Expand file tree Collapse file tree 3 files changed +16
-0
lines changed
Expand file tree Collapse file tree 3 files changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -8,6 +8,7 @@ python_library(
88 "//executorch/backends/arm:tosa_quant_utils",
99 "//executorch/backends/arm:tosa_utils",
1010 "//executorch/backends/xnnpack/_passes:xnnpack_passes",
11+ "//executorch/backends/transforms:replace_scalar_with_tensor",
1112 "//executorch/exir:lib",
1213 ],
1314)
Original file line number Diff line number Diff line change @@ -256,6 +256,7 @@ python_library(
256256 "//executorch/backends/cadence/aot:pass_utils",
257257 "//executorch/backends/cadence/aot:remove_ops",
258258 "//executorch/backends/cadence/aot:utils",
259+ "//executorch/backends/transforms:replace_scalar_with_tensor",
259260 "//executorch/exir:pass_base",
260261 "//executorch/exir/dialects:lib",
261262 "//executorch/exir/dialects/edge:lib",
Original file line number Diff line number Diff line change @@ -201,6 +201,20 @@ def define_common_targets():
201201 ],
202202 )
203203
204+ runtime .python_library (
205+ name = "replace_scalar_with_tensor" ,
206+ srcs = [
207+ "replace_scalar_with_tensor.py" ,
208+ ],
209+ visibility = [
210+ "//executorch/backends/..." ,
211+ ],
212+ deps = [
213+ "//caffe2:torch" ,
214+ "//executorch/exir:pass_base" ,
215+ ],
216+ )
217+
204218 runtime .python_test (
205219 name = "test_duplicate_dynamic_quant_chain" ,
206220 srcs = [
You can’t perform that action at this time.
0 commit comments