[LoweringStrategy] Use a more general method to fetch input dims and sizes

[LoweringStrategy] Use a more general method to fetch input dims and sizes #298