| 28 | |
| 29 | template<typename To, typename Ti> |
| 30 | Array<To> cplx(const Array<Ti> &lhs, const Array<Ti> &rhs, |
| 31 | const af::dim4 &odims) { |
| 32 | common::Node_ptr lhs_node = lhs.getNode(); |
| 33 | common::Node_ptr rhs_node = rhs.getNode(); |
| 34 | |
| 35 | jit::BinaryNode<To, Ti, af_cplx2_t> *node = |
| 36 | new jit::BinaryNode<To, Ti, af_cplx2_t>(lhs_node, rhs_node); |
| 37 | |
| 38 | return createNodeArray<To>(odims, common::Node_ptr(node)); |
| 39 | } |
| 40 | |
| 41 | #define CPLX_UNARY_FN(op) \ |
| 42 | template<typename To, typename Ti> \ |