Casts layer norm's input to scale's dtype; otherwise bf16 inference graph generation complains.
Casts layer norm's input to scale's dtype; otherwise bf16 inference graph
generation complains.

PiperOrigin-RevId: 335562268

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