megatron.data.dataset\_utils.get\_a\_and\_b\_segments ===================================================== .. currentmodule:: megatron.data.dataset_utils .. autofunction:: get_a_and_b_segments