Add support for `add_metric` in the graph function mode.
- raise error if name is not provided for metric in eager mode. - increased test coverage for add_metric API to include graph function use case and added a correctness test. PiperOrigin-RevId: 234690946
Loading
Please sign in to comment