Skip to content
Snippets Groups Projects
Commit 4ed46d9b authored by Galen Andrew's avatar Galen Andrew Committed by tensorflow-copybara
Browse files

Add measurements for ClippingFactory, ZeroingFactory and ZeroingClippingFactory.

For now measurements are aggregated by SumFactory, but we will add support in the future for private measurements aggregation.

Also:
  1. Change state to OrderedDict (from tuple).
  2. Change default zeroing norm order from np.inf to 2 for ease of transition between ZeroingFactory and ZeroingClippingFactory (which must use 2-norm), and use tf.linalg.global_norm if norm order is 2, for efficiency.
PiperOrigin-RevId: 339908232
parent 4f9caffb
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment