megatron.data.dataset_utils.get_a_and_b_segments# megatron.data.dataset_utils.get_a_and_b_segments(sample, np_rng)# Divide sample into a and b segments.