| load("@fbcode_macros//build_defs:python_library.bzl", "python_library") |
| |
| oncall("executorch") |
| |
| python_library( |
| name = "xnnpack_passes", |
| srcs = [ |
| "__init__.py", |
| "channels_last_tagged_reshape_pass.py", |
| "conv1d_unsqueeze_pass.py", |
| "convert_to_linear.py", |
| "convert_to_sdpa.py", |
| "convert_to_upsample_bilinear2d.py", |
| "fuse_activation_pass.py", |
| "fuse_batch_norm_with_conv.py", |
| "prelu_reshape_pass.py", |
| "remove_getitem_op.py", |
| "tag_implicit_q_dq_pass.py", |
| "xnnpack_pass.py", |
| ], |
| deps = [ |
| "//caffe2:torch", |
| "//executorch/backends/transforms:addmm_mm_to_linear", |
| "//executorch/backends/transforms:lib", |
| "//executorch/backends/xnnpack/partition:configs", |
| "//executorch/backends/xnnpack/partition:partitioner_graphs", |
| "//executorch/backends/xnnpack/serialization:xnnpack_schema", |
| "//executorch/backends/xnnpack/utils:xnnpack_utils", |
| "//executorch/exir:lib", |
| "//executorch/exir:pass_base", |
| "//executorch/exir/dialects:lib", |
| "//executorch/exir/passes:const_prop_pass", |
| "//executorch/exir/passes:memory_format_ops_pass", |
| "//executorch/exir/program:program", |
| ], |
| ) |
