Searched refs:values_lib (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_utils.py | 24 from tensorflow.python.distribute import values as values_lib unknown
35 def regroup(values, wrap_class=values_lib.PerReplica, always_wrap=False):
98 if same_id and isinstance(v0, values_lib.DistributedVariable):
118 assert not isinstance(v0, values_lib.MirroredVariable), (
137 if (isinstance(x, values_lib.DistributedVariable) or
138 not isinstance(x, values_lib.DistributedValues)):
156 if isinstance(x, values_lib.DistributedValues) and not is_mirrored(x):
167 regrouped = regroup(updates, values_lib.Mirrored)
173 return values_lib.Mirrored(values)
192 return values_lib.Mirrored(with_dep)
[all …]
|
| D | values_test.py | 40 from tensorflow.python.distribute import values as values_lib unknown
82 return values_lib.Mirrored(v)
100 var_cls = values_lib.MirroredVariable
333 v = values_lib.DistributedDelegate((Foo(7), Foo(8)))
340 v = values_lib.DistributedDelegate((7, 8))
390 v = values_lib.DistributedDelegate((Foo(7), Foo(8)))
605 value = values_lib.PerReplica([1., 2.])
958 mirrored = values_lib.MirroredVariable(
1118 var_cls = values_lib.SyncOnReadVariable
1165 replica_local = values_lib.SyncOnReadVariable(
[all …]
|