Function AssignKernel

paddle/phi/kernels/selected_rows/assign_kernel.cc:25–31

// as input if needed
template <typename Context>
void AssignKernel(const Context& dev_ctx,
                  const SelectedRows& x,
                  SelectedRows* out) {
  out->set_rows(x.rows());
  out->set_height(x.height());
  phi::AssignKernel<Context>(dev_ctx, x.value(), out->mutable_value());
}

}  // namespace phi::sr