ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Classes | Namespaces | Functions
linalg_op.h File Reference
#include <ATen/core/tensor.h>
#include <ATen/ops/einsum_op.h>
Include dependency graph for linalg_op.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  container::op::add_op
 A functor to perform add operation on a Tensor. More...
 
struct  container::op::mul_op
 
struct  container::op::div_op
 
struct  container::op::transpose_op< Conjugate >
 
struct  container::op::stride_op
 A functor to perform stride operation on a Tensor. More...
 
struct  container::op::inflate_op
 A functor for inflating a tensor. More...
 
struct  container::op::reduce_op
 

Namespaces

namespace  container
 
namespace  container::op
 

Functions

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::Tensoroperator+= (ct::Tensor &self, const ct::Tensor &other)
 
ct::Tensoroperator-= (ct::Tensor &self, const ct::Tensor &other)
 
ct::Tensoroperator*= (ct::Tensor &self, const ct::Tensor &other)
 
ct::Tensoroperator/= (ct::Tensor &self, const ct::Tensor &other)
 

Function Documentation

◆ operator*()

ct::Tensor operator* ( const ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator*=()

ct::Tensor & operator*= ( ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator+()

ct::Tensor operator+ ( const ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator+=()

ct::Tensor & operator+= ( ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator-()

ct::Tensor operator- ( const ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator-=()

ct::Tensor & operator-= ( ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator/()

ct::Tensor operator/ ( const ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function:

◆ operator/=()

ct::Tensor & operator/= ( ct::Tensor self,
const ct::Tensor other 
)
Here is the call graph for this function: