message: Improve the shape function for ParameterizedTruncatedNormal
  The parameters of ParameterizedTruncatedNormal should
  be 0-D or 1-D, which is checked in ther kernel functions.
  There is no check in the shape function of the ops.
  This fix improves the shape function and checks the
  parameters of ParameterizedTruncatedNormal whever possible.
diff: @@ -72,7 +72,15 @@ REGISTER_OP("ParameterizedTruncatedNormal")
  .Attr("seed2: int = 0")
  .Attr("dtype: {half,bfloat16,float,double}")
  .Attr("T: {int32, int64}")
  - .SetShapeFn(shape_inference::RandomShape);
  + .SetShapeFn([](InferenceContext* c) {
  + ShapeHandle unused;
  + // Parameters must be 0-d or 1-d.
  + TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused));
  + TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused));
  + TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(3), 1, &unused));
  + TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &unused));
  + return shape_inference::RandomShape(c);
  + });
  REGISTER_OP("TruncatedNormal")
  .Input("shape: T")
  