ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
Classes | |
struct | add_op |
A functor to perform add operation on a Tensor. More... | |
struct | div_op |
class | EinsumOpTest |
struct | inflate_op |
A functor for inflating a tensor. More... | |
class | LinalgOpTest |
struct | mul_op |
struct | reduce_op |
struct | stride_op |
A functor to perform stride operation on a Tensor. More... | |
struct | transpose_op |
Functions | |
template<typename... Tensors> | |
std::enable_if< std::is_same< typenamestd::common_type< Tensors... >::type, Tensor >::value, Tensor >::type | einsum_impl (const std::string &equation, const EinsumOption &option, const Tensors &... tensors) |
Computes the Einstein summation convention on the given tensors based on the expression passed as a string. | |
TYPED_TEST_SUITE (EinsumOpTest, base::utils::Types) | |
TYPED_TEST (EinsumOpTest, Transform) | |
TYPED_TEST (EinsumOpTest, Reduce) | |
TYPED_TEST (EinsumOpTest, Stride) | |
TYPED_TEST (EinsumOpTest, Inflate) | |
TYPED_TEST (EinsumOpTest, ContractDot) | |
TYPED_TEST (EinsumOpTest, ContractGemv) | |
TYPED_TEST (EinsumOpTest, ContractGemm) | |
TYPED_TEST (EinsumOpTest, TransformEllipsis) | |
TYPED_TEST (EinsumOpTest, ReduceEllipsis) | |
TYPED_TEST (EinsumOpTest, StrideEllipsis) | |
TYPED_TEST (EinsumOpTest, InflateEllipsis) | |
TYPED_TEST (EinsumOpTest, ContractGemmEllipsis) | |
TYPED_TEST_SUITE (LinalgOpTest, base::utils::Types) | |
TYPED_TEST (LinalgOpTest, Add) | |
TYPED_TEST (LinalgOpTest, Sub) | |
TYPED_TEST (LinalgOpTest, AddScalar) | |
TYPED_TEST (LinalgOpTest, Mul) | |
TYPED_TEST (LinalgOpTest, MulScalar) | |
TYPED_TEST (LinalgOpTest, Div) | |
TYPED_TEST (LinalgOpTest, Transpose) | |
TYPED_TEST (LinalgOpTest, Stride) | |
TYPED_TEST (LinalgOpTest, Inflate) | |
TYPED_TEST (LinalgOpTest, Reduce) | |
std::enable_if< std::is_same< typenamestd::common_type< Tensors... >::type, Tensor >::value, Tensor >::type container::op::einsum_impl | ( | const std::string & | equation, |
const EinsumOption & | option, | ||
const Tensors &... | tensors | ||
) |
Computes the Einstein summation convention on the given tensors based on the expression passed as a string.
Tensors | Variadic template parameter pack for the input tensors. |
equation | The Einstein summation convention expression. |
tensors | The input tensors for the summation operation. |
std::invalid_argument | if the expression or the input tensors are invalid. |
std::runtime_error | if an error occurs while performing the summation operation. |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
ContractDot | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
ContractGemm | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
ContractGemmEllipsis | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
ContractGemv | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
Inflate | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
InflateEllipsis | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
Reduce | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
ReduceEllipsis | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
Stride | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
StrideEllipsis | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
Transform | |||
) |
container::op::TYPED_TEST | ( | EinsumOpTest | , |
TransformEllipsis | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Add | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
AddScalar | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Div | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Inflate | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Mul | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
MulScalar | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Reduce | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Stride | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Sub | |||
) |
container::op::TYPED_TEST | ( | LinalgOpTest | , |
Transpose | |||
) |
container::op::TYPED_TEST_SUITE | ( | EinsumOpTest | , |
base::utils::Types | |||
) |
container::op::TYPED_TEST_SUITE | ( | LinalgOpTest | , |
base::utils::Types | |||
) |