Searched refs:lhs_replicated (Results 1 – 1 of 1) sorted by relevance
1542 auto lhs_replicated = lhs.Reshard(HloSharding::Replicate()).hlo(); in PartitionBaseCase() local1543 TF_ASSIGN_OR_RETURN(auto dot, create_sharded_dot(lhs_replicated, rhs.hlo(), in PartitionBaseCase()