Package | Description |
---|---|
org.tensorflow.op.core |
Modifier and Type | Method and Description |
---|---|
static CollectiveGather.Options |
CollectiveGather.communicationHint(String communicationHint) |
CollectiveGather.Options |
CollectiveGather.Options.communicationHint(String communicationHint) |
static CollectiveGather.Options |
CollectiveGather.communicationHint(String communicationHint) |
CollectiveGather.Options |
CollectiveGather.Options.communicationHint(String communicationHint) |
Modifier and Type | Method and Description |
---|---|
static <T extends Number> |
CollectiveGather.create(Scope scope,
Operand<T> input,
Long groupSize,
Long groupKey,
Long instanceKey,
Shape shape,
CollectiveGather.Options... options)
Factory method to create a class wrapping a new CollectiveGather operation.
|
static <T extends Number> |
CollectiveGather.create(Scope scope,
Operand<T> input,
Long groupSize,
Long groupKey,
Long instanceKey,
Shape shape,
CollectiveGather.Options... options)
Factory method to create a class wrapping a new CollectiveGather operation.
|
Copyright © 2022. All rights reserved.