Function TransDataLayout

paddle/fluid/framework/data_layout_transform.cc:53–69

}

void TransDataLayout(const phi::KernelKey& kernel_type_for_var,
                     const phi::KernelKey& expected_kernel_type,
                     const phi::DenseTensor& in,
                     phi::DenseTensor* out,
                     const phi::Place& place) {
  PADDLE_ENFORCE(
      backends_are_same_class(kernel_type_for_var.backend(),
                              expected_kernel_type.backend()),
      common::errors::PreconditionNotMet(
          "TransDataLayout only support DataLayout transform on same place."));

  TransDataLayout(kernel_type_for_var.layout(),
                  expected_kernel_type.layout(),
                  place,
                  in,
                  out);
}

void TransDataLayout(DataLayout from_layout,
                     DataLayout to_layout,