operation_tags.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/linalg/contraction_axes.hpp>
#include <uni20/linalg/reduction_axes.hpp>
 
#include <cstddef>
#include <string_view>
#include <type_traits>
 
namespace uni20::linalg
{
 
struct least_squares_op
{
    static constexpr std::string_view name = "least_squares";
};
 
struct svd_least_squares_op
{
    static constexpr std::string_view name = "svd_least_squares";
};
 
struct copy_op
{
    static constexpr std::string_view name = "copy";
};
 
struct conjugate_inplace_op
{
    static constexpr std::string_view name = "conjugate_inplace";
};
 
template <class Function> struct transform_op
{
    static constexpr std::string_view name = "transform";
    [[no_unique_address]] Function function;
};
 
template <class Function> transform_op(Function) -> transform_op<std::decay_t<Function>>;
 
template <class Function> struct transform_inplace_op
{
    static constexpr std::string_view name = "transform_inplace";
    [[no_unique_address]] Function function;
};
 
template <class Function> transform_inplace_op(Function) -> transform_inplace_op<std::decay_t<Function>>;
 
struct assign_product_op
{
    static constexpr std::string_view name = "assign_product";
};
 
struct gemm_op
{
    static constexpr std::string_view name = "gemm";
};
 
struct gemv_op
{
    static constexpr std::string_view name = "gemv";
};
 
template <std::size_t LhsRank, std::size_t RhsRank, std::size_t ContractedRank> struct contract_op
{
    static constexpr std::string_view name = "contract";
    ContractionAxes<LhsRank, RhsRank, ContractedRank> axes;
};
 
struct inner_product_op
{
    static constexpr std::string_view name = "inner_product";
};
 
struct norm_op
{
    static constexpr std::string_view name = "norm";
};
 
enum class MatrixNorm
{
  MaxAbs,
  One,
  Infinity,
  Frobenius
};
 
struct matrix_norm_op
{
    static constexpr std::string_view name = "matrix_norm";
    MatrixNorm kind;
};
 
struct lu_rcond_op
{
    static constexpr std::string_view name = "lu_rcond";
    MatrixNorm norm = MatrixNorm::One;
};
 
struct linear_solve_op
{
    static constexpr std::string_view name = "linear_solve";
};
 
struct lu_factor_op
{
    static constexpr std::string_view name = "lu_factor";
};
 
struct lu_solve_op
{
    static constexpr std::string_view name = "lu_solve";
};
 
struct qr_op
{
    static constexpr std::string_view name = "qr";
};
 
struct lq_op
{
    static constexpr std::string_view name = "lq";
};
 
template <std::size_t InputRank, std::size_t ReducedRank> struct sum_reduction_op
{
    static constexpr std::string_view name = "sum";
    ReductionAxes<InputRank, ReducedRank> axes;
};
 
struct matrix_exponential_op
{
    static constexpr std::string_view name = "matrix_exponential";
};
 
enum class MatrixRegion
{
  All,
  Upper,
  Lower
};
 
struct matrix_set_op
{
    static constexpr std::string_view name = "matrix_set";
    MatrixRegion region = MatrixRegion::All;
};
 
enum class MatrixTriangle
{
  Upper,
  Lower
};
 
struct SelfAdjointEighOptions
{
    bool compute_vectors = true;
    MatrixTriangle triangle = MatrixTriangle::Upper;
};
 
struct self_adjoint_eigh_op
{
    static constexpr std::string_view name = "self_adjoint_eigh";
    bool compute_vectors = true;
    MatrixTriangle triangle = MatrixTriangle::Upper;
};
 
enum class SvdVectorExtent
{
  Reduced,
  Full
};
 
struct SvdOptions
{
    SvdVectorExtent left = SvdVectorExtent::Reduced;
    SvdVectorExtent right = SvdVectorExtent::Reduced;
};
 
enum class SvdOverwrite
{
  None,
  Left,
  Right
};
 
struct singular_values_op
{
    static constexpr std::string_view name = "singular_values";
};
 
struct svd_left_op
{
    static constexpr std::string_view name = "svd_left";
    SvdVectorExtent left = SvdVectorExtent::Reduced;
    bool overwrite_input = false;
};
 
struct svd_right_op
{
    static constexpr std::string_view name = "svd_right";
    SvdVectorExtent right = SvdVectorExtent::Reduced;
    bool overwrite_input = false;
};
 
struct svd_op
{
    static constexpr std::string_view name = "svd";
    SvdVectorExtent left = SvdVectorExtent::Reduced;
    SvdVectorExtent right = SvdVectorExtent::Reduced;
    SvdOverwrite overwrite = SvdOverwrite::None;
};
 
struct symmetric_tridiagonal_eigen_op
{
    static constexpr std::string_view name = "symmetric_tridiagonal_eigen";
    bool compute_vectors = true;
};
 
struct nonsymmetric_eigen_op
{
    static constexpr std::string_view name = "nonsymmetric_eigen";
    bool compute_right_vectors = true;
};
 
struct schur_op
{
    static constexpr std::string_view name = "schur";
    bool compute_vectors = true;
};
 
struct hessenberg_schur_op
{
    static constexpr std::string_view name = "hessenberg_schur";
    bool compute_vectors = true;
};
 
struct schur_reorder_op
{
    static constexpr std::string_view name = "reorder_schur";
    std::size_t from = 0;
    std::size_t to = 0;
    bool update_vectors = true;
};
 
} // namespace uni20::linalg