Commit 3ea28fa0 authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Wrap all stateless metrics functions given in `compile`/`add_metric` APIs with... 

Wrap all stateless metrics functions given in `compile`/`add_metric` APIs with a stateful metric for fit and evaluate APIs

- added support for the new metrics in TPUs. Thank you xiejw@ for helping me with this.
- fixed weighted averaging issue with metrics.
- refactored the code to be simpler as all metrics and losses that we log are stateful now.

PiperOrigin-RevId: 219885890
parent 8a91e6ad
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