blob: a2445f26c0dfe349f82e4648be652797ebec5420 [file]
load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
python_library(
name = "arm_quantizer",
srcs = ["arm_quantizer.py"],
deps = [
":arm_quantizer_utils",
"//caffe2:torch",
"//executorch/backends/arm/quantizer/quantization_annotation:quantization_annotation",
"//executorch/exir:lib",
],
)
python_library(
name = "quantization_config",
srcs = ["quantization_config.py"],
deps = [
"//caffe2:torch",
],
)
python_library(
name = "arm_quantizer_utils",
srcs = ["arm_quantizer_utils.py"],
deps = [
":quantization_config",
],
)