model input has different size for dims and reshape for reshape_variable0