message: Support all numeric type in XLA all-to-all from TensorFlow.
diff: @@ -26,7 +26,7 @@ REGISTER_OP("AllToAll")
  .Input("input: T")
  .Input("group_assignment: int32")
  .Output("output: T")
  - .Attr("T: {bfloat16, float}")
  + .Attr("T: {numbertype, bool}")
  .Attr("concat_dimension: int")
  .Attr("split_dimension: int")
  .Attr("split_count: int")
  