Function iterator_function

dali/test/python/jax_plugin/test_iterator_decorator.py:41–42

()

    # given
    @data_iterator(output_map=["data"], reader_name="reader")
    def iterator_function():
        return iterator_function_def()

    iter = iterator_function(num_threads=4, device_id=0, batch_size=batch_size)