| 21 | |
| 22 | template<typename T> |
| 23 | Array<T> reorder(const Array<T> &in, const af::dim4 &rdims) { |
| 24 | const af::dim4 &iDims = in.dims(); |
| 25 | af::dim4 oDims(0); |
| 26 | for (int i = 0; i < 4; i++) { oDims[i] = iDims[rdims[i]]; } |
| 27 | |
| 28 | Array<T> out = createEmptyArray<T>(oDims); |
| 29 | getQueue().enqueue(kernel::reorder<T>, out, in, oDims, rdims); |
| 30 | return out; |
| 31 | } |
| 32 | |
| 33 | #define INSTANTIATE(T) \ |
| 34 | template Array<T> reorder<T>(const Array<T> &in, const af::dim4 &rdims); |
no test coverage detected