Method _prep_input_sets

dali/python/nvidia/dali/_debug_mode.py:573–593

(self, inputs)

                raise_err("ndim", actual_ndim, expected_ndim)

    def _prep_input_sets(self, inputs):
        inputs = list(inputs)

        for i, input in enumerate(inputs):
            # Transforming any convertible datatype to
            # TensorList (DataNodeDebugs are already unpacked).
            # Additionally accepting input sets, but only as list of TensorList.
            if not isinstance(input, (_tensors.TensorListCPU, _tensors.TensorListGPU)) and not (
                isinstance(input, list)
                and all(
                    [
                        isinstance(elem, (_tensors.TensorListCPU, _tensors.TensorListGPU))
                        for elem in input
                    ]
                )
            ):
                inputs[i] = _transform_data_to_tensorlist(
                    input, len(input), device_id=self._device_id
                )

        return _build_input_sets(inputs, self._op_name)

    def _update_classification(self, old_collection, position, new_classification):
        """Keeps the data classification up to date in case of running the conditional mode or