only compute max eigenvector decomposition if we are computing the max eigenvector
only compute max eigenvector decomposition if we are computing the max eigenvector

PiperOrigin-RevId: 455701401

google/init2winit