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