Class | Description |
---|---|
AllReduce<T extends Number> |
Mutually reduces multiple tensors of identical type and shape.
|
AllReduce.Options |
Optional attributes for
AllReduce |
BroadcastRecv<T extends Number> |
Receives a tensor value broadcast from another device.
|
BroadcastRecv.Options |
Optional attributes for
BroadcastRecv |
BroadcastSend<T extends Number> |
Broadcasts a tensor value to one or more other devices.
|
BroadcastSend.Options |
Optional attributes for
BroadcastSend |
Copyright © 2022. All rights reserved.