Make a couple functions in sharding propagation externally visible fo…

…r use in auto-sharding (in following CLs). PiperOrigin-RevId: 578402519
Intel-tensorflow · Nov 1, 2023 · 7330702 · 7330702
1 parent 8e67b5d
commit 7330702
Show file tree

Hide file tree

Showing 3 changed files with 220 additions and 203 deletions.
diff --git a/third_party/xla/xla/service/BUILD b/third_party/xla/xla/service/BUILD
@@ -666,6 +666,7 @@ cc_library(
         ":dot_as_convolution_util",
         ":hlo_graph_dumper",
         ":hlo_pass",
+        "//xla:array",
         "//xla:protobuf_util",
         "//xla:shape_tree",
         "//xla:shape_util",