message: Support XLA cross-replica sum on integer types from TensorFlow.
diff: @@ -76,7 +76,7 @@ REGISTER_OP("CrossReplicaSum")
  .Input("input: T")
  .Input("group_assignment: int32")
  .Output("output: T")
  - .Attr("T: {bfloat16, float}")
  + .Attr("T: {bfloat16, float, int32, uint32}")
  .SetShapeFn(shape_inference::UnchangedShape);
  REGISTER_OP("CollectivePermute")
  