Searched refs:rhs_mat (Results 1 – 3 of 3) sorted by relevance
123 MatrixMap rhs_mat(rhs.data(), k, n); in TEST() local127 matmul_result = lhs_mat * rhs_mat; in TEST()139 sum += std::abs(lhs_mat(i % m, k1) * rhs_mat(k1, i / m)); in TEST()
148 EigenMatrixFloatMap rhs_mat(output_rhs_tensor->matrix<float>().data(), in Compute() local150 rhs_mat.setZero(); in Compute()271 rhs_mat.col(input_index) += in Compute()
797 rhs_mat = array_ops.expand_dims(rhs, axis=-1)798 solution_mat = self.solve(rhs_mat, adjoint=adjoint)