Function iterator_function

dali/test/python/jax_plugin/test_multigpu.py:403–404

(shard_id, num_shards)

        reader_name="reader",
    )
    def iterator_function(shard_id, num_shards):
        return iterator_function_def(shard_id=shard_id, num_shards=num_shards)

    data_iterator_instance = iterator_function(
        batch_size=batch_size,