def make_dataset(time_us, num_elements):
    return dataset_ops.Dataset.range(num_elements).apply(sleep.sleep(time_us))
def make_dataset(time_us, num_elements):
    return dataset_ops.Dataset.range(num_elements).apply(sleep.sleep(time_us))