diff options
| author |  Pavithra Vijay <psv@google.com> | 2018-06-29 18:02:18 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-29 18:05:25 -0700 |
| commit | c290930ec1beacbcac414b43b3367dd44ffbd303 (patch) | |
| tree | b1136e7c32718a6f1f9ebfde3073c88546078de6 /third_party | |
| parent | a520735d205ca5678fc8a371ea1add00413907fe (diff) | |
Add `synchronization` and `aggregation` args to get_variable(). These args will be used for distributed variables.
Add Enum `VariableSynchronization` with values for `synchronization`: AUTO, UNREPLICATED, ON_WRITE, ON_READ
Add Enum `VariableAggregation` with values for `aggregation`: NONE, SUM, MEAN. Replace all the aggregation methods strings in distribution strategy to the enum values.
Update Mirrored strategy to use these parameters to decide on whether a variable should be Mirrored or TowerLocal.
Update different distribution strategy value types to use the `VariableAggregation` Enum
PiperOrigin-RevId: 202736077
Diffstat (limited to 'third_party')
0 files changed, 0 insertions, 0 deletions