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

CollectiveAssignGroupV2

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

Assign group keys based on group assignment.

Public Methods

static CollectiveAssignGroupV2
create(Scope scope, Operand<Integer> groupAssignment, Operand<Integer> deviceIndex, Operand<Integer> baseKey)
Factory method to create a class wrapping a new CollectiveAssignGroupV2 operation.
Output<Integer>
Output<Integer>

Inherited Methods

Public Methods

public static CollectiveAssignGroupV2 create (Scope scope, Operand<Integer> groupAssignment, Operand<Integer> deviceIndex, Operand<Integer> baseKey)

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

Parameters
scope current scope
Returns
  • a new instance of CollectiveAssignGroupV2

public Output<Integer> groupKey ()

public Output<Integer> groupSize ()