Method check_input

dali/python/nvidia/dali/_debug_mode.py:328–335

(self, other_size, other_context, op_name, input_idx)

        return False

    def check_input(self, other_size, other_context, op_name, input_idx):
        if not self.set_if_empty(other_size, other_context) and self.size != other_size:
            raise RuntimeError(
                f"Batch size must be uniform across an iteration. "
                f"Input {input_idx} for operator '{op_name}' has batch "
                f"size = {other_size}. Expected batch size = {self.size}"
                f"from:\n{self._source_context}"
            )

    def check_external_source(self, other_size, other_context, output_idx=-1):
        if not self.set_if_empty(other_size, other_context) and self.size != other_size: