|
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 | |||
| ) |