Update logic to rescale L2 loss in distribution strategy (#6601)
Update logic to rescale L2 loss in distribution strategy (#6601)


tombstone/models,alexgorban/models,alexgorban/models,derekjchow/models,tombstone/models,alexgorban/models,derekjchow/models,tombstone/models,tombstone/models,derekjchow/models,alexgorban/models,tombstone/models,tombstone/models,derekjchow/models,derekjchow/models,alexgorban/models