| 41 | |
| 42 | template<typename T> |
| 43 | Array<T> moddimOp(const Array<T> &in, af::dim4 outDim) { |
| 44 | |
| 45 | const auto &node = in.getNode(); |
| 46 | |
| 47 | NodeIterator<> it(node.get()); |
| 48 | |
| 49 | dim4 olddims_t = in.dims(); |
| 50 | |
| 51 | bool all_linear = true; |
| 52 | while (all_linear && it != NodeIterator<>()) { |
| 53 | all_linear &= it->isLinear(olddims_t.get()); |
| 54 | ++it; |
| 55 | } |
| 56 | if (all_linear == false) in.eval(); |
| 57 | |
| 58 | Array<T> out = createNodeArray<T>(outDim, copyModdims(in.getNode(), outDim)); |
| 59 | |
| 60 | return out; |
| 61 | } |
| 62 | |
| 63 | template<typename T> |
| 64 | Array<T> modDims(const Array<T> &in, const af::dim4 &newDims) { |