Route iterator_function

dali/test/python/jax_plugin/test_multigpu.py:None–None

data_iterator(
        output_map=output_map,
        sharding=sharding,
        last_batch_policy=LastBatchPolicy.DROP,
        reader_name="reader",
    )

# Copyright (c) 2023-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.