Searched refs:padded_target_shape (Results 1 – 1 of 1) sorted by relevance
870 Shape padded_target_shape = shard_shape; in ReplicatePartial() local872 padded_target_shape.set_dimensions( in ReplicatePartial()887 HloInstruction::CreateBroadcast(padded_target_shape, zero, {})); in ReplicatePartial()888 auto offsets = MakePartitionOffsets(padded_target_shape, sharding(), in ReplicatePartial()892 padded_target_shape, zero_bcast, hlo_, offsets)); in ReplicatePartial()899 if (!ShapeUtil::Compatible(target_shape, padded_target_shape)) { in ReplicatePartial()