CollectiveAssignGroupV2

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 ()