Searched refs:ReplicatePartial (Results 1 – 2 of 2) sorted by relevance
341 HloInstruction* ReplicatePartial(absl::Span<const int64> dims);
861 HloInstruction* result = ReplicatePartial(all_dims); in Replicate()866 HloInstruction* PartitionedHlo::ReplicatePartial(absl::Span<const int64> dims) { in ReplicatePartial() function in xla::spmd::PartitionedHlo969 partial_replicate_hlo.ReplicatePartial(replicate_dims); in ReshardToPartialReplicateWithAllGather()