| 176 | } |
| 177 | |
| 178 | void dns_dns_mul(int m, int n, int k, const SyncArray<kernel_type> &dense_mat, |
| 179 | const SyncArray<kernel_type> &origin_dense, SyncArray<kernel_type> &result){ |
| 180 | Eigen::Map<const Eigen::Matrix<kernel_type, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor>> denseMat(dense_mat.host_data(), k, n); |
| 181 | Eigen::Map<const Eigen::Matrix<kernel_type, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>> |
| 182 | originDenseMat(origin_dense.host_data(), m, k); |
| 183 | Eigen::Matrix<kernel_type, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor> retMat = originDenseMat * denseMat; |
| 184 | Eigen::Map<Eigen::Matrix<kernel_type, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor> >(result.host_data(), |
| 185 | retMat.rows(), |
| 186 | retMat.cols()) = retMat; |
| 187 | } |
| 188 | } |
no test coverage detected