Function _process_argument_inputs

dali/python/nvidia/dali/ops/__init__.py:273–304

Process argument inputs: validate, deprecate, promote constants, add to spec and return the list of DataNodes that need to be kept alive for the graph to exist. Returns ------- List the list of DataNodes representing inputs (there may be conversions)

(schema, spec, kwargs, operator_name)

def _process_argument_inputs(schema, spec, kwargs, operator_name):
    """Process argument inputs: validate, deprecate, promote constants, add to spec and return the
    list of DataNodes that need to be kept alive for the graph to exist.

    Returns
    -------
    List
        the list of DataNodes representing inputs (there may be conversions)
    """
    kwargs = _handle_arg_deprecations(schema, kwargs, operator_name)
    # Add argument inputs
    result = []
    for k in sorted(kwargs.keys()):
        arg_inp = kwargs[k]
        if arg_inp is None:
            continue
        # Argument input constants are always placed on CPU
        # TODO(klecki): For MIS we can extract this step outside and do it once
        arg_inp = _handle_constant(arg_inp, "cpu", k, operator_name)

        if arg_inp.device != "cpu":
            raise ValueError(
                f"Invalid device \"{arg_inp.device}\" for argument '{k}' of operator "
                f"'{operator_name}'. "
                f"Named arguments must be on CPU."
            )

        _check_arg_input(schema, operator_name, k)

        spec.AddArgumentInput(k, arg_inp.name, arg_inp.ndim, arg_inp.dtype, arg_inp.layout)
        result.append(arg_inp)
    return result


def _process_inputs(schema, spec, inputs, operator_name):