message: Add shape inference function for XlaScatter
diff: @@ -762,7 +762,7 @@ REGISTER_OP("XlaScatter")
  .Attr("T: numbertype")
  .Attr("Tindices: {int32, int64}")
  .Output("output: T")
  - .SetShapeFn(UnchangedRank)
  + .SetShapeFn(shape_inference::UnchangedShape)
  .Doc(R"doc(
  Wraps the XLA Scatter operator documented at
  https://www.tensorflow.org/xla/operation_semantics#scatter.
  