-
Sean Augenstein authored
Tweaks keras_utils to allow specifying a single loss with multiple model outputs and/or labels, where the outputs/labels are pushed directly into the (probably custom user-defined) loss. PiperOrigin-RevId: 347030237
5721e8f9