|
TR-mbed 1.0
|
GeneralScalarContraction is a template class that provides the scalar value of Tensor -Tensor contraction operation, when all the dimensions are contracting dimensions. This Kernel reduces two tensors to an scalar. More...
#include <TensorContractionSycl.h>
Public Types | |
| typedef cl::sycl::accessor< OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local > | Scratch |
Public Member Functions | |
| EIGEN_DEVICE_FUNC | GeneralScalarContraction (Scratch scratch_, const LhsMapper lhs_, const RhsMapper rhs_, OutAccessor out_res_, const StorageIndex rng_) |
| EIGEN_DEVICE_FUNC void | operator() (cl::sycl::nd_item< 1 > itemID) |
Public Attributes | |
| Scratch | scratch |
| const LhsMapper | lhs |
| const RhsMapper | rhs |
| OutAccessor | out_res |
| const StorageIndex | rng |
GeneralScalarContraction is a template class that provides the scalar value of Tensor -Tensor contraction operation, when all the dimensions are contracting dimensions. This Kernel reduces two tensors to an scalar.
| OutScalar | determines the output scalar type |
| LhsScalar | determines the left-hand-side scalar type |
| RhsScalar | determines the right-hand-side scalar type |
| OutAccessor | determines the sycl accessor type for out put (please see the sycl-1.2.1 specification (https://www.khronos.org/registry/SYCL/specs/sycl-1.2.1.pdf) for accessor definition) |
| LhsMapper | determines the tensor contraction mapper type for left-hand-side matrix |
| RhsMapper | determines the tensor contraction mapper type for right-hand-side matrix |
| StorageIndex | determines the StorageIndex Type |
| Vectorizable | determines whether or not the vectorization is enabled for the Eigen expression. |
| scratch | local memory containing tiles of LHS and RHS tensors for each work-group |
| lhs | determines the left-hand-side flattened tensor (tensor mapper) |
| rhs | determines the right-hand-side flattened tensor (tensor mapper) |
| out_res | determines the output tensor containing the contraction result |
| rng | determins the total input data size |
| typedef cl::sycl::accessor<OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local> Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::Scratch |
|
inline |
|
inline |
| const LhsMapper Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::lhs |
| OutAccessor Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::out_res |
| const RhsMapper Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::rhs |
| const StorageIndex Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::rng |
| Scratch Eigen::TensorSycl::internal::GeneralScalarContraction< OutScalar, LhsScalar, RhsScalar, OutAccessor, LhsMapper, RhsMapper, StorageIndex, Vectorizable >::scratch |