Commit 8930d5af authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Adding `add_metric` API for layers/models in Keras. 

Supports
- adding metric tensor (both aggregated and not aggregated) on model/layer in graph mode.
- adding metric tensor (both aggregated and not aggregated) in model/layer call in graph and eager mode.

PiperOrigin-RevId: 221865769
parent dd18b138
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