Function StrictReduceIntArrayAxisInferMetaBase

paddle/phi/infermeta/unary.cc:4344–4359

}

void StrictReduceIntArrayAxisInferMetaBase(const MetaTensor& x,
                                           const IntArray& axis,
                                           bool keep_dim,
                                           bool reduce_all,
                                           MetaTensor* out,
                                           MetaConfig config) {
  DDim out_dim;
  if (config.is_runtime || !axis.FromTensor()) {
    out_dim = StrictReduceInferDim(x, axis.GetData(), keep_dim, reduce_all);
  } else {
    out_dim = ReduceInferDimForIntArrayAxis(x, axis, keep_dim, reduce_all);
  }
  out->set_dims(out_dim);
  out->set_dtype(x.dtype());
  out->set_layout(x.layout());
}

void StrictReduceIntArrayAxisInferMeta(const MetaTensor& x,
                                       const IntArray& axis,