Internal code change

PiperOrigin-RevId: 631243354
tensorflow · May 7, 2024 · 2a945eb · 2a945eb
1 parent 35fd24c
commit 2a945eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/core/framework/BUILD b/tensorflow/core/framework/BUILD
@@ -716,6 +716,7 @@ cc_library(
         "//learning/deepmind/tensorflow/queues:__pkg__",
         "//learning/deepmind/tensorflow/sstable:__pkg__",
         "//tensorflow/compiler/mlir/tools/kernel_gen:__pkg__",
+        "//third_party/py/grain/_src/tensorflow/ops:__pkg__",
         "//waymo/ml/compiler/frontend/kernels:__pkg__",
     ],
     deps = [