| 12 | } |
| 13 | |
| 14 | void EigenSolver::compute_batch_symmetric(size_t dim, VectorF matrices) { |
| 15 | if (dim == 2) { |
| 16 | compute_batch_symmetric_2x2(matrices); |
| 17 | } else if (dim == 3) { |
| 18 | compute_batch_symmetric_3x3(matrices); |
| 19 | } else { |
| 20 | std::stringstream err_msg; |
| 21 | err_msg << "EigenSolver does not support " << dim << "D matrix yet."; |
| 22 | throw NotImplementedError(err_msg.str()); |
| 23 | } |
| 24 | } |
| 25 | |
| 26 | void EigenSolver::compute_batch_symmetric_2x2(const VectorF& matrices) { |
| 27 | const size_t dim = 2; |