Function StrictReduceIntArrayAxisInferMeta

paddle/phi/infermeta/unary.cc:4361–4372

}

void StrictReduceIntArrayAxisInferMeta(const MetaTensor& x,
                                       const IntArray& axis,
                                       bool keep_dim,
                                       MetaTensor* out,
                                       MetaConfig config) {
  bool reduce_all = false;
  if (axis.size() == 0) {
    reduce_all = true;
  }
  StrictReduceIntArrayAxisInferMetaBase(
      x, axis, keep_dim, reduce_all, out, config);
}

void ReduceScatterInferMeta(const MetaTensor& x, int nranks, MetaTensor* out) {
  auto dim = x.dims();