Package | Description |
---|---|
org.tensorflow.op.collective |
Modifier and Type | Method and Description |
---|---|
static CollectiveAllToAll.Options |
CollectiveAllToAll.timeoutSeconds(Float timeoutSeconds)
Sets the timeoutSeconds option.
|
CollectiveAllToAll.Options |
CollectiveAllToAll.Options.timeoutSeconds(Float timeoutSeconds)
Sets the timeoutSeconds option.
|
Modifier and Type | Method and Description |
---|---|
static <T extends TNumber> |
CollectiveAllToAll.create(Scope scope,
Operand<T> input,
Operand<? extends TType> communicator,
Operand<TInt32> groupAssignment,
CollectiveAllToAll.Options... options)
Factory method to create a class wrapping a new CollectiveAllToAllV3 operation.
|
Copyright © 2015–2022. All rights reserved.