Assign group keys based on group assignment.
tf.raw_ops.CollectiveAssignGroupV2(
group_assignment, device_index, base_key, name=None
)
Args | |
---|---|
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). |
Returns | |
---|---|
A tuple of Tensor objects (group_size, group_key).
|
|
group_size
|
A Tensor of type int32 .
|
group_key
|
A Tensor of type int32 .
|