MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / transpose

Function transpose

dnn/src/cuda/svd/opr_impl.cpp:31–36  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

29}
30
31void transpose(
32 megdnn::cuda::HandleImpl* handle, const TensorND& src, const TensorND& dst) {
33 TensorLayout t = transposed_layout(src.layout);
34 megdnn_assert(t.total_nr_elems() == dst.layout.total_nr_elems());
35 handle->relayout_opr()->exec({src.raw_ptr(), t}, dst);
36}
37
38} // namespace
39

Callers 1

execMethod · 0.70

Calls 5

transposed_layoutFunction · 0.85
total_nr_elemsMethod · 0.45
execMethod · 0.45
relayout_oprMethod · 0.45
raw_ptrMethod · 0.45

Tested by

no test coverage detected