    def offsets_struct_type(self) -> str:
        """The type of the offsets struct variable used in runtime."""
        num_jagged_dims = len(self.jagged_dims())
        return f"ait::JaggedOffsets<{self.offsets_type()}, {num_jagged_dims}>"
The type of the offsets struct variable used in runtime.
python/aitemplate/compiler/base.py