tf.raw_ops.CollectiveAssignGroupV2

Assign group keys based on group assignment.

group_assignmentA Tensor of type int32.
device_indexA Tensor of type int32.
base_keyA Tensor of type int32.
nameA name for the operation (optional).

A tuple of Tensor objects (group_size, group_key).
group_sizeA Tensor of type int32.
group_keyA Tensor of type int32.