Commit 02dfd621 authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Wraps losses passed to the `compile` API (strings and v1 losses) which are not... 

Wraps losses passed to the `compile` API (strings and v1 losses) which are not instances of v2 `Loss` class in `LossWrapper` class. => All losses will now use `SUM_OVER_BATCH_SIZE` reduction as default.

PiperOrigin-RevId: 230995424
parent b8898d89
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment