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

Function transpose

src/backend/cuda/transpose.cpp:23–36  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21
22template<typename T>
23Array<T> transpose(const Array<T> &in, const bool conjugate) {
24 const dim4 &inDims = in.dims();
25
26 dim4 outDims = dim4(inDims[1], inDims[0], inDims[2], inDims[3]);
27
28 Array<T> out = createEmptyArray<T>(outDims);
29
30 const bool is32multiple =
31 inDims[0] % kernel::TILE_DIM == 0 && inDims[1] % kernel::TILE_DIM == 0;
32
33 kernel::transpose<T>(out, in, conjugate, is32multiple);
34
35 return out;
36}
37
38#define INSTANTIATE(T) \
39 template Array<T> transpose(const Array<T> &in, const bool conjugate);

Callers 1

svdFunction · 0.70

Calls 2

dim4Class · 0.50
dimsMethod · 0.45

Tested by

no test coverage detected