Function run_pipeline

dali/test/python/test_utils_tensorflow.py:259–270

(pipelines, iterations, device, to_stop_iter=False)

def run_pipeline(pipelines, iterations, device, to_stop_iter=False):
    if not isinstance(pipelines, list):
        pipelines = [pipelines]
    results = []
    with expect_iter_end(not to_stop_iter, StopIteration):
        for _ in range(iterations):
            shard_outputs = []
            for pipeline in pipelines:
                pipe_outputs = pipeline.run()
                shard_outputs.append(tuple(to_array(result) for result in pipe_outputs))
            results.append(tuple(shard_outputs))
    return results


def compare(dataset_results, standalone_results, iterations=-1, num_devices=1):