Package | Description |
---|---|
org.tensorflow.op.distribute |
Class and Description |
---|
NcclAllReduce
Outputs a tensor containing the reduction across all input tensors.
|
NcclBroadcast
Sends
input to all devices that are connected to the output. |
NcclReduce
Reduces
input from num_devices using reduction to a single device. |
Copyright © 2015–2022. All rights reserved.