Commit 855d2a72 authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Add `synchronization` and `aggregation` args to the layer `add_weight()` API.... 

Add `synchronization` and `aggregation` args to the layer `add_weight()` API. These args will be used for distributed variables.

Migrate all usages of `tower_local_var_scope` to using the new args.

PiperOrigin-RevId: 203855963
parent e0541b71
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