Function run_dataset_eager_mode

dali/test/python/test_utils_tensorflow.py:248–256

(dali_datasets, iterations, to_stop_iter=False)

def run_dataset_eager_mode(dali_datasets, iterations, to_stop_iter=False):
    if not isinstance(dali_datasets, list):
        dali_datasets = [dali_datasets]

    results = []
    with expect_iter_end(not to_stop_iter, StopIteration):
        for i, batch in zip(range(iterations), zip(*dali_datasets)):
            results.append(batch)
    return results


def run_pipeline(pipelines, iterations, device, to_stop_iter=False):