    def get_max_dense_shape(self) -> List[IntVar]:
        """
        Returns a list of IntVars representing the maximum dense shape
        (rectangular volume) that the JaggedIntVar can correspond to.
        The result has the batch_dim as the first item and the IntImm
        with the max_value of each JaggedDim that follows.
        """
        result = [self.batch_dim()]
        for dim in self.jagged_dims():
            result.append(dim.max_value())
        return result
Returns a list of IntVars representing the maximum dense shape
(rectangular volume) that the JaggedIntVar can correspond to.
The result has the batch_dim as the first item and the IntImm
with the max_value of each JaggedDim that follows.
python/aitemplate/compiler/base.py