Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

CollectiveAllToAllV3

Stay organized with collections Save and categorize content based on your preferences.
public final class CollectiveAllToAllV3

Mutually exchanges multiple tensors of identical type and shape.

Nested Classes

class CollectiveAllToAllV3.Options Optional attributes for CollectiveAllToAllV3  

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static <T extends Number> CollectiveAllToAllV3<T>
create(Scope scope, Operand<T> input, Operand<?> communicator, Operand<Integer> groupAssignment, Options... options)
Factory method to create a class wrapping a new CollectiveAllToAllV3 operation.
Output<T>
data()
static CollectiveAllToAllV3.Options
timeoutSeconds(Float timeoutSeconds)

Inherited Methods

Public Methods

public Output<T> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static CollectiveAllToAllV3<T> create (Scope scope, Operand<T> input, Operand<?> communicator, Operand<Integer> groupAssignment, Options... options)

Factory method to create a class wrapping a new CollectiveAllToAllV3 operation.

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of CollectiveAllToAllV3

public Output<T> data ()

public static CollectiveAllToAllV3.Options timeoutSeconds (Float timeoutSeconds)