tf.raw_ops.CollectiveAssignGroupV2

Assign group keys based on group assignment.

group_assignment A Tensor of type int32.
device_index A Tensor of type int32.
base_key A Tensor of type int32.
name A name for the operation (optional).

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