pytorch

1
def define_targets(rules):
2
    rules.py_library(
3
        name = "autograd",
4
        srcs = rules.glob(["*.py"]),
5
        data = rules.glob([
6
            "*.yaml",
7
            "templates/*",
8
        ]),
9
        visibility = ["//:__subpackages__"],
10
        deps = [
11
            rules.requirement("PyYAML"),
12
            "//torchgen",
13
        ],
14
    )
15