MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / PerformTranspose

Function PerformTranspose

paddle/phi/kernels/impl/einsum_kernel_impl.h:477–491  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

475
476template <typename T, typename Context>
477DenseTensor PerformTranspose(const Context& dev_ctx,
478 const DenseTensor& tensor,
479 const LabelMap& label2perm,
480 const std::vector<char>& all_labels,
481 const LabelMap& label2type) {
482 auto axis = GetLabelIndexByType<int>(
483 all_labels, label2type, label2perm, LabelType::ALL_TYPE);
484 VLOG(5) << "PerformTranspose: " << paddle::string::join_strings(axis, ",");
485 if (is_no_need_transpose(axis)) {
486 return tensor;
487 }
488 auto ret = Transpose<T, Context>(dev_ctx, tensor, axis);
489 VLOG(5) << "PerformTranspose: do_transpose()";
490 return ret;
491}
492
493template <typename T, typename Context>
494DenseTensor PerformContraction(

Callers

nothing calls this directly

Calls 2

join_stringsFunction · 0.85
is_no_need_transposeFunction · 0.85

Tested by

no test coverage detected