model input has different size for dims and reshape for reshape_elementcount1