Function TransferLayout

paddle/phi/kernels/transfer_layout_kernel.h:31–41

template <typename Context>
DenseTensor TransferLayout(const Context& dev_ctx,
                           const DenseTensor& x,
                           DataLayout dst_layout) {
  DenseTensor dense_out = Empty(dev_ctx, {x.dtype(), x.dims(), dst_layout});
  TransferLayoutKernel<Context>(dev_ctx,
                                x,
                                static_cast<int>(x.layout()),
                                static_cast<int>(dst_layout),
                                &dense_out);
  return dense_out;
}

}  // namespace phi