| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 200 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
203 all_values = per_replica_value.values
220 def _simple_gather(per_replica_value, reduce_to_device, axis): argument
222 all_values = per_replica_value.values
255 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
285 if not isinstance(per_replica_value, value_lib.DistributedValues):
286 per_replica_value = _make_tensor_into_per_replica(per_replica_value)
292 per_replica_value.values) == 1 and _devices_match(
293 per_replica_value, destinations):
294 with ops.device(per_replica_value.values[0].device):
[all …]
|
| D | strategy_gather_test.py | 279 per_replica_value = strategy.experimental_distribute_values_from_function(
282 def replica_fn(per_replica_value): argument
284 local_value = array_ops.identity(per_replica_value)
291 strategy.run(replica_fn, args=(per_replica_value,)))
354 def replica_fn(per_replica_value): argument
356 return ctx.all_gather(array_ops.identity(per_replica_value), axis=axis)
374 per_replica_value = strategy.experimental_distribute_values_from_function(
390 strategy.run(run, args=(per_replica_value,)))
402 per_replica_value = strategy.experimental_distribute_values_from_function(
418 strategy.run(run, args=(per_replica_value,)))
[all …]
|
| D | cross_device_ops_test.py | 243 per_replica_value = make_per_replica_value(value_fn, devices)
244 reduced_values = collective.reduce(options.reduce_op, per_replica_value,
245 per_replica_value,
288 per_replica_value = make_per_replica_value(value_fn, devices)
289 value_dst_pairs.append((per_replica_value, per_replica_value))
650 per_replica_value = make_per_replica_value(value, devices)
652 per_replica_value, per_replica_value, axis=axis, options=options)
|
| D | strategy_common_test.py | 80 per_replica_value = strategy.experimental_distribute_values_from_function(
86 reduce_util.ReduceOp.SUM, value=per_replica_value, axis=None)
|
| D | vars_test.py | 178 per_replica_value = values.PerReplica(
195 updates = [("assign", per_replica_value), ("assign_add", per_replica_value),
761 per_replica_value = values.PerReplica(
775 updates = [("assign", per_replica_value)]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
|