https://github.com/ponder-lab/ML/blob/bea52a494f5dcf0c1e56ef7f13c03b9b7c83d1c0/com.ibm.wala.cast.python.test/data/tf2_test_ragged_constant17.py#L22 Currently, we are inferring (3, 3) instead of (2, 3) for the inner shape.