Function SquareKernel

paddle/phi/kernels/selected_rows/activation_kernel.cc:26–32

template <typename T, typename Context>
void SquareKernel(const Context& dev_ctx,
                  const SelectedRows& x,
                  SelectedRows* out) {
  out->set_rows(x.rows());
  out->set_height(x.height());
  phi::SquareKernel<T, Context>(dev_ctx, x.value(), out->mutable_value());
}

template <typename T, typename Context>
void SqrtKernel(const Context& dev_ctx,