linalg_op.h

Go to the documentation of this file.

#ifndef ATEN_OPS_LINALG_H_
#define ATEN_OPS_LINALG_H_
 
#include <ATen/core/tensor.h>
 
#include <ATen/ops/einsum_op.h>
 
namespace container {
namespace op {
 
struct add_op {
    void operator()(
        const Tensor& x,
        const Tensor& y,
        Tensor& z);
 
    template <typename T>
    void operator()(
        const T& alpha,
        const Tensor& x,
        const T& beta,
        const Tensor& y,
        Tensor& z);
};
 
struct mul_op {
    // z = x * y
    void operator()(
        const Tensor& x,
        const Tensor& y,
        Tensor& z);
 
    // y = alpha * x
    template <typename T>
    void operator()(
        const T& alpha,
        const Tensor& x,
        Tensor& y);
};
 
struct div_op {
    // z = x / y
    void operator()(
        const Tensor& x,
        const Tensor& y,
        Tensor& z);
};
 
template <bool Conjugate = false>
struct transpose_op {
    void operator()(
        const Tensor& input,
        const std::vector<int>& permutation,
        Tensor& output);
};
 
struct stride_op {
    void operator()(
        const Tensor& input,
        const std::vector<int64_t>& stride,
        Tensor& output);
};
 
struct inflate_op {
    void operator()(
        const Tensor& input,
        const std::vector<int64_t>& stride,
        Tensor& output);
};
 
 
struct reduce_op {
    void operator()(
        const Tensor& input,
        const int64_t& inner_most_dim,
        Tensor& output);
};
 
} // namespace op
} // namespace container
 
ct::Tensor   operator+(const ct::Tensor& self, const ct::Tensor& other);
ct::Tensor   operator-(const ct::Tensor& self, const ct::Tensor& other);
ct::Tensor   operator*(const ct::Tensor& self, const ct::Tensor& other);
ct::Tensor   operator/(const ct::Tensor& self, const ct::Tensor& other);
ct::Tensor& operator+=(ct::Tensor& self, const ct::Tensor& other);
ct::Tensor& operator-=(ct::Tensor& self, const ct::Tensor& other);
ct::Tensor& operator*=(ct::Tensor& self, const ct::Tensor& other);
ct::Tensor& operator/=(ct::Tensor& self, const ct::Tensor& other);
 
#endif // ATEN_OPS_LINALG_H_