-
Andrew M. Dai authored
Change logit and activation shapes to follow standard convention with batch size in the first dimension. PiperOrigin-RevId: 172172495
ed54e8dd
Change logit and activation shapes to follow standard convention with batch size in the first dimension. PiperOrigin-RevId: 172172495