Function ReduceCheckIfOneDNNSupport

paddle/phi/kernels/reduce_kernel_impl.cc:57–63

}

bool ReduceCheckIfOneDNNSupport(const KernelContext* dev_ctx) {
  if (dev_ctx->InputAt<DenseTensor>(0).dims().size() > 5 ||
      !HasOptimizedOneDNNKernel(dev_ctx)) {
    return false;
  }
  return true;
}

bool ReduceMeanCheckIfOneDNNSupport(const KernelContext* dev_ctx) {
  std::cout << "ReduceMeanCheckIfOneDNNSupport" << std::endl;