Function pipeline

dali/test/python/test_external_source_dali.py:482–490

()

    @pipeline_def
    def pipeline():
        outputs = fn.external_source(
            source=sources[source_type],
            num_outputs=num_outputs,
            batch=source_type != "sample_cb_source",
        )
        assert len(outputs) == num_outputs
        utilized_outputs = (out for out, is_used in zip(outputs, usage_mask) if is_used)
        return tuple(fn.gaussian_blur(out, window_size=3) for out in utilized_outputs)

    pipe = pipeline(batch_size=max_batch_size, num_threads=4, device_id=0)
    unused_output_idxs = [i for i, is_used in enumerate(usage_mask) if not is_used]