Function set_constant

paddle/phi/kernels/funcs/math_function.cc:277–300

};

void set_constant(const DeviceContext& dev_ctx,
                  DenseTensor* tensor,
                  float value) {
  TensorSetConstantWithPlace func(dev_ctx, tensor, value);
#ifdef PADDLE_WITH_CUSTOM_DEVICE
  if (dev_ctx.GetPlace().GetType() == AllocationType::CUSTOM) {
    func(CustomPlace());
    return;
  }
#endif
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
  // tensor->place().apply_visitor(func);
  phi::VisitPlace(tensor->place(), func);
#elif defined(PADDLE_WITH_XPU)
  if (dev_ctx.GetPlace().GetType() == AllocationType::XPU) {
    func(phi::XPUPlace());
    return;
  } else {
    func(CPUPlace());
  }
#else
  func(CPUPlace());
#endif
}

template struct ColwiseSum<CPUContext, float>;
template struct ColwiseSum<CPUContext, double>;