| 27 | |
| 28 | template<typename T> |
| 29 | Array<T> matmul(const Array<T> &lhs, const Array<T> &rhs, af_mat_prop optLhs, |
| 30 | af_mat_prop optRhs) { |
| 31 | int Mdim = optLhs == AF_MAT_NONE ? 0 : 1; |
| 32 | int Ndim = optRhs == AF_MAT_NONE ? 1 : 0; |
| 33 | Array<T> res = createEmptyArray<T>( |
| 34 | dim4(lhs.dims()[Mdim], rhs.dims()[Ndim], lhs.dims()[2], lhs.dims()[3])); |
| 35 | static const T alpha = T(1.0); |
| 36 | static const T beta = T(0.0); |
| 37 | gemm(res, optLhs, optRhs, &alpha, lhs, rhs, &beta); |
| 38 | return res; |
| 39 | } |
| 40 | |
| 41 | template<typename T> |
| 42 | Array<T> dot(const Array<T> &lhs, const Array<T> &rhs, af_mat_prop optLhs, |
no test coverage detected