diff --git a/BUILD.bazel b/BUILD.bazel
index 79bf1b4018b..2327db6564f 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -1906,12 +1906,20 @@ test_suite(
 # as if they are from this package, so we alias them for
 # compatibility.
 
-alias(
-    name = "native_functions.yaml",
-    actual = "aten/src/ATen/native/native_functions.yaml",
-)
-
-alias(
-    name = "tags.yaml",
-    actual = "aten/src/ATen/native/tags.yaml",
-)
+[
+    alias(
+        name = paths.basename(path),
+        actual = path,
+    )
+    for path in [
+        "aten/src/ATen/templates/DispatchKeyNativeFunctions.cpp",
+        "aten/src/ATen/templates/DispatchKeyNativeFunctions.h",
+        "aten/src/ATen/templates/LazyIr.h",
+        "aten/src/ATen/templates/RegisterDispatchKey.cpp",
+        "aten/src/ATen/native/native_functions.yaml",
+        "aten/src/ATen/native/tags.yaml",
+        "aten/src/ATen/native/ts_native_functions.yaml",
+        "torch/csrc/lazy/core/shape_inference.h",
+        "torch/csrc/lazy/ts_backend/ts_native_functions.cpp",
+    ]
+]
diff --git a/build.bzl b/build.bzl
index 6302d55b844..d4d3a5898f5 100644
--- a/build.bzl
+++ b/build.bzl
@@ -25,15 +25,15 @@ def define_targets(rules):
     rules.genrule(
         name = "generate-code",
         srcs = [
-            "aten/src/ATen/native/ts_native_functions.yaml",
-            "aten/src/ATen/templates/DispatchKeyNativeFunctions.cpp",
-            "aten/src/ATen/templates/DispatchKeyNativeFunctions.h",
-            "aten/src/ATen/templates/LazyIr.h",
-            "aten/src/ATen/templates/RegisterDispatchKey.cpp",
-            "torch/csrc/lazy/core/shape_inference.h",
-            "torch/csrc/lazy/ts_backend/ts_native_functions.cpp",
+            ":DispatchKeyNativeFunctions.cpp",
+            ":DispatchKeyNativeFunctions.h",
+            ":LazyIr.h",
+            ":RegisterDispatchKey.cpp",
             ":native_functions.yaml",
+            ":shape_inference.h",
             ":tags.yaml",
+            ":ts_native_functions.cpp",
+            ":ts_native_functions.yaml",
         ],
         tools = ["//tools/setup_helpers:generate_code"],
         outs = _GENERATED_CPP + GENERATED_AUTOGRAD_H + GENERATED_LAZY_H + GENERATED_TESTING_PY,