# tf.compat.v1.tpu.cross_replica_sum

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.14.0/tensorflow/python/tpu/ops/tpu_ops.py#L88-L105) |

Sum the input tensor across replicas according to group_assignment.

    tf.compat.v1.tpu.cross_replica_sum(
        x, group_assignment=None, name=None
    )

<br />

<br />

<br />

| ## Args ||
|---|---|
| `x` | The local tensor to the sum. |
| `group_assignment` | Optional 2d int32 lists with shape \[num_groups, num_replicas_per_group\]. `group_assignment[i]` represents the replica ids in the ith subgroup. |
| `name` | Optional op name. |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| A `Tensor` which is summed across replicas. ||

<br />