message: [XLA] Allow different output shape than inputs for convolution in hlo_evaluator.
  Convolution allows generally to have a different output shape than the shapes
  in the input, but hlo_evaluator doesn't allow it. Relaxing the restriction.
diff: @@ -1287,11 +1287,12 @@ class HloEvaluatorTypedVisitor : public DfsHloVisitorWithDefault {
  CHECK_EQ(num_spatial_dims + 2, lhs_rank);
  CHECK_EQ(num_spatial_dims + 2, rhs_rank);
  - TF_ASSIGN_OR_RETURN(auto inferred_return_shape,
  + TF_ASSIGN_OR_RETURN(
  + auto inferred_return_shape,
  ShapeInference::InferConvolveShape(
  lhs_shape, rhs_shape, conv->feature_group_count(),
  conv->batch_group_count(), window, dnums,
  - /*preferred_element_type=*/absl::nullopt));
  + /*preferred_element_type=*/conv->shape().element_type()));
  CHECK(ShapeUtil::Compatible(result_shape, inferred_return_shape))
  << "return shape set to: " << ShapeUtil::HumanString(result_shape)
  << " but is inferred to be: "
  