Searched refs:PerReplica (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 104 if isinstance(input_tensor, value_lib.PerReplica):
114 return value_lib.PerReplica(device_map, (input_tensor,))
145 if not all(isinstance(v[0], value_lib.PerReplica)
250 if not isinstance(per_replica_value, value_lib.PerReplica):
|
| D | values.py | 383 class PerReplica(DistributedValues): class
1228 class SyncOnReadVariable(DistributedVariable, PerReplica, trackable.Trackable):
1300 def regroup(device_map, values, wrap_class=PerReplica):
|
| D | parameter_server_strategy.py | 407 elif isinstance(x, values.PerReplica):
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 68 return value_lib.PerReplica(value_lib.ReplicaDeviceMap(devices), index)
295 per_replica = value_lib.PerReplica(
330 per_replica = value_lib.PerReplica(
|
| D | cross_device_utils_test.py | 107 per_replica = value_lib.PerReplica(device_map, (t0, t1))
|
| D | mirrored_strategy_multigpu_test.py | 1392 factors = values.PerReplica(device_map, (5.0, 3.0))
1393 expected_result = values.PerReplica(device_map, (5.0 * 1.25, 3.0 * 1.25))
|
| D | values_test.py | 193 def _is_per_replica(self, result, expected, klass=values.PerReplica):
|