Function pipeline

dali/test/python/test_dltensor_operator.py:406–411

()

    @pipeline_def(batch_size=len(input_batch), num_threads=4, device_id=0)
    def pipeline():
        data = fn.external_source(lambda: input_batch)
        data = fn.dl_tensor_python_function(
            data.gpu(), batch_processing=True, function=sliced_tensor, synchronize_stream=False
        )
        return data

    p = pipeline()
    (out,) = p.run()