Function get_input_arg_per_sample

dali/test/python/sequences_test_utils.py:317–324

(input_data, param_cb, rng)

def get_input_arg_per_sample(input_data, param_cb, rng):
    return [
        [
            param_cb(SampleDesc(rng, None, sample_idx, batch_idx, sample))
            for sample_idx, sample in enumerate(batch)
        ]
        for batch_idx, batch in enumerate(input_data.data)
    ]


def get_input_arg_per_frame(input_data: ArgData, param_cb, rng, check_broadcasting):