Home
last modified time | relevance | path

Searched refs:value_lib (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py37 from tensorflow.python.distribute import values as value_lib unknown
73 (value_lib.DistributedValues, ops.Tensor, ps_values.AggregatingVariable,
86 if isinstance(value, value_lib.DistributedValues):
118 if isinstance(input_tensor, value_lib.PerReplica):
121 return value_lib.PerReplica((input_tensor,))
155 if not all(isinstance(v[0], value_lib.PerReplica)
164 if isinstance(destinations, value_lib.DistributedValues):
197 wrap_class=value_lib.Mirrored)
285 if not isinstance(per_replica_value, value_lib.DistributedValues):
296 return distribute_utils.regroup((v,), wrap_class=value_lib.Mirrored)
[all …]
Dcross_device_utils.py24 from tensorflow.python.distribute import values as value_lib unknown
606 assert isinstance(value, value_lib.DistributedValues)
Dcross_device_ops_test.py40 from tensorflow.python.distribute import values as value_lib unknown
92 return value_lib.PerReplica(values)
205 elif isinstance(value, value_lib.Mirrored):
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py41 from tensorflow.python.distribute import values as value_lib unknown
56 elif isinstance(devices, value_lib.DistributedValues):
96 wrap_class=value_lib.Mirrored)
113 wrap_class=value_lib.Mirrored)
143 if isinstance(left, value_lib.Mirrored):
243 per_replica = value_lib.PerReplica((t0, t1))
372 per_replica = value_lib.PerReplica((t0, t1))
662 per_replica = value_lib.PerReplica(indexed_slices)
702 self.assertIsInstance(result, value_lib.Mirrored)