xref: /aosp_15_r20/external/executorch/exir/dialects/edge/op/TARGETS (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
2load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
3
4oncall("ai_infra_mobile_platform")
5
6python_library(
7    name = "lib",
8    srcs = [
9        "api.py",
10        "sample_input.py",
11    ],
12    deps = [
13        "//caffe2:torch",
14        "//caffe2/torchgen:torchgen",
15        "//executorch/exir/dialects/edge/arg:lib",
16        "//executorch/exir/operator:convert",
17    ],
18)
19
20python_unittest(
21    name = "test_api",
22    srcs = [
23        "test/test_api.py",
24    ],
25    deps = [
26        ":lib",
27        "//caffe2:torch",
28        "//caffe2/torchgen:torchgen",
29    ],
30)
31