def dataset_fn():
    return dataset_ops.Dataset.range(1).repeat().interleave(_make_fake_dataset_fn(), cycle_length=10, num_parallel_calls=dataset_ops.AUTOTUNE)
def dataset_fn():
    return dataset_ops.Dataset.range(1).repeat().interleave(_make_fake_dataset_fn(), cycle_length=10, num_parallel_calls=dataset_ops.AUTOTUNE)