Extract a common helper function `HandleElementwiseWithDimsToReplicate` in `SpmdPartitioningVisitor`.#21233

Merged

copybara-service[bot] merged 1 commit intomainfrom test_713827953

Jan 11, 2025

+92-16

Commits on Jan 11, 2025

Extract a common helper function HandleElementwiseWithDimsToReplicate in SpmdPartitioningVisitor.

ZixuanJiang
authored and
Google-ML-Automation
committed