def _is_jagged_shape(shape: List[IntVar]) -> bool:
    """Whether the given shape is a shape of a jagged Tensor."""
    return len(shape) > 0 and isinstance(shape[0], JaggedIntVar)
Whether the given shape is a shape of a jagged Tensor.
python/aitemplate/backend/common/elementwise_common.py