Also returns per_example_loss in ComputeLoss() function in image classifiers.
Also returns per_example_loss in ComputeLoss() function in image classifiers.

PiperOrigin-RevId: 273599207

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo