message: Check in shape function that axis and shift are same size
diff: @@ -36,7 +36,8 @@ REGISTER_OP("Roll")
  TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused));
  // The `axis` must be scalar or 1-D.
  TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused));
  -
  + // Validate 'shift' is the same shape as axis'.
  + TF_RETURN_IF_ERROR(c->Merge(c->input(1), c->input(2), &unused) );
  return shape_inference::UnchangedShape(c);
  });
  