This makes it easier (smaller CL diff) to merge output size configuration and initialize() in a follow-up. PiperOrigin-RevId: 438543247
