@@ -10,16 +10,17 @@ runtime.python_library(
1010 ],
1111)
1212
13- # Exposed through __init__.py
1413runtime.python_library(
1514 name = "arm_quantizer",
1615 srcs = ["arm_quantizer.py"],
1716 deps = [
1817 ":arm_quantizer_utils",
1918 ":quantization_annotator",
19+ ":quantizer_support",
2020 "//executorch/backends/arm:constants",
2121 "//executorch/backends/arm:ethosu",
2222 "//executorch/backends/arm:vgf",
23+ "//executorch/backends/cortex_m/quantizer:quantizer",
2324 "//executorch/backends/arm/tosa:specification",
2425 "//executorch/backends/arm:arm_compile_spec",
2526 "//caffe2:torch",
@@ -43,11 +44,24 @@ runtime.python_library(
4344 name = "arm_quantizer_utils",
4445 srcs = ["arm_quantizer_utils.py"],
4546 deps = [
47+ "//caffe2:torch",
48+ "//executorch/backends/arm:common",
49+ "//executorch/backends/arm:constants",
4650 ":quantization_config",
4751 "//pytorch/ao:torchao",
4852 ],
4953)
5054
55+ runtime.python_library(
56+ name = "quantizer_support",
57+ srcs = ["quantizer_support.py"],
58+ deps = [
59+ ":quantization_annotator",
60+ "//caffe2:torch",
61+ "//executorch/backends/cortex_m/quantizer:quantizer",
62+ ],
63+ )
64+
5165runtime.python_library(
5266 name = "lib",
5367 srcs = ["__init__.py"],
0 commit comments