Searched refs:per_replica (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 168 per_replica = _make_per_replica(values, devices)
191 per_replica,
203 per_replica,
217 [(per_replica, d1),
223 [(per_replica, d1),
243 per_replica = value_lib.PerReplica((t0, t1))
247 reduce_op, [(per_replica, per_replica)])
249 result = cross_device_ops_instance.reduce(reduce_op, per_replica,
250 per_replica)
372 per_replica = value_lib.PerReplica((t0, t1))
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_test.py | 191 per_replica = []
193 per_replica.append(val * ctx.replica_id_in_sync_group)
194 return tuple(per_replica)
1388 per_replica = values_lib.PerReplica(vals)
1390 spec = per_replica._type_spec
1397 per_replica = values_lib.PerReplica(vals)
1399 spec = per_replica._type_spec
1400 tensor_list = spec._to_components(per_replica)
1403 self.assertAllEqual(per_replica.values, reconstructed.values)
1408 per_replica = values_lib.PerReplica(vals)
[all …]
|
| D | cross_device_ops.py | 143 per_replica = _make_tensor_into_per_replica(pair[0])
144 result.append((per_replica, pair[1]))
1224 for per_replica in reversed(per_replica_values):
1226 values_by_device[i].append(per_replica.values[i])
1285 for per_replica in per_replica_values:
1289 per_replica.values[i], implementation, options.timeout_seconds))
1353 for per_replica in per_replica_values:
1357 per_replica.values[i], axis, implementation,
|