message: Add axis shape check for tf.roll
diff: @@ -32,6 +32,8 @@ REGISTER_OP("Roll")
  shape_inference::ShapeHandle unused;
  // The `input` must be 1-D or higher
  TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused));
  + // The `axis` must be scalar or 1-D.
  + TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused));
  return shape_inference::UnchangedShape(c);
  });
  