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