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