org.platanios.tensorflow.api.ops.training.distribute.ops
Intermediate device to reduce to. If None
, reduce to the first device in the provided destination
of the reduce()
method.
Reduces a batch per-device values to the corresponding provided destinations.
Reduces a batch per-device values to the corresponding provided destinations.
For each tuple in valueDestinationPairs
, this method reduces each first element to each second element which
indicates the corresponding destination.
Reduction method.
Sequence of per-device values and destinations pairs. If a destination is None
,
then the destination is set to match the devices of the corresponding per-device
value.
Sequence of reduced mirrored values.
Broadcasts value
to destination
.
Broadcasts value
to destination
.
Value to broadcast.
Broadcast destination.
Broadcasted mirrored value.
Intermediate device to reduce to.
Intermediate device to reduce to. If None
, reduce to the first device in the provided destination
of the reduce()
method.
Reduces value
to destination
.
Reduces value
to destination
.
It runs the reduction operation defined by reduction
and puts the result on destination
.
Reduction method.
Per-device value to be reduced.
Reduction destination.
Reduced mirrored value.
Cross-tower ops that always perform a reduction to one device first and then do broadcasting.
Batch reduction is done by reduction on each element one by one.