| 58 | } |
| 59 | |
| 60 | af_array gforReorder(const af_array in, unsigned dim) { |
| 61 | // This is here to stop gcc from complaining |
| 62 | if (dim > 3) { AF_THROW_ERR("GFor: Dimension is invalid", AF_ERR_SIZE); } |
| 63 | unsigned order[AF_MAX_DIMS] = {0, 1, 2, dim}; |
| 64 | |
| 65 | order[dim] = 3; |
| 66 | af_array out; |
| 67 | AF_THROW(af_reorder(&out, in, order[0], order[1], order[2], order[3])); |
| 68 | return out; |
| 69 | } |
| 70 | |
| 71 | af::dim4 seqToDims(af_index_t *indices, af::dim4 parentDims, |
| 72 | bool reorder = true) { |