Class | Description |
---|---|
CollectiveAllToAll<T extends TNumber> |
Mutually exchanges multiple tensors of identical type and shape.
|
CollectiveAllToAll.Inputs<T extends TNumber> | |
CollectiveAllToAll.Options |
Optional attributes for
CollectiveAllToAll |
CollectiveBcastRecv<U extends TType> |
Receives a tensor value broadcast from another device.
|
CollectiveBcastRecv.Inputs | |
CollectiveBcastRecv.Options |
Optional attributes for
CollectiveBcastRecv |
CollectiveBcastSend<T extends TType> |
Broadcasts a tensor value to one or more other devices.
|
CollectiveBcastSend.Inputs<T extends TType> | |
CollectiveBcastSend.Options |
Optional attributes for
CollectiveBcastSend |
CollectiveGather<T extends TNumber> |
Mutually accumulates multiple tensors of identical type and shape.
|
CollectiveGather.Inputs<T extends TNumber> | |
CollectiveGather.Options |
Optional attributes for
CollectiveGather |
CollectiveInitializeCommunicator |
Initializes a group for collective operations.
|
CollectiveInitializeCommunicator.Inputs | |
CollectiveInitializeCommunicator.Options |
Optional attributes for
CollectiveInitializeCommunicator |
CollectivePermute<T extends TType> |
An Op to permute tensors across replicated TPU instances.
|
CollectivePermute.Inputs<T extends TType> | |
CollectiveReduce<T extends TNumber> |
Mutually reduces multiple tensors of identical type and shape.
|
CollectiveReduce.Inputs<T extends TNumber> | |
CollectiveReduce.Options |
Optional attributes for
CollectiveReduce |
Copyright © 2015–2022. All rights reserved.