#include <TensorReductionSycl.h>
Public Types | |
typedef Evaluator::CoeffReturnType | CoeffReturnType |
typedef Evaluator::EvaluatorPointerType | EvaluatorPointerType |
typedef Evaluator::Index | Index |
typedef OpDef::type | Op |
typedef OpDefiner< OpType, CoeffReturnType, Index, false > | OpDef |
typedef cl::sycl::accessor< CoeffReturnType, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local > | ScratchAcc |
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | element_wise_reduce (Index globalRId, Index globalPId, CoeffReturnType &accumulator) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | operator() (cl::sycl::nd_item< 1 > itemID) |
PartialReductionKernel (ScratchAcc scratch_, Evaluator evaluator_, EvaluatorPointerType output_accessor_, OpType op_, const Index preserve_elements_num_groups_, const Index reduce_elements_num_groups_, const Index num_coeffs_to_preserve_, const Index num_coeffs_to_reduce_) | |
Public Attributes | |
Evaluator | evaluator |
const Index | num_coeffs_to_preserve |
const Index | num_coeffs_to_reduce |
Op | op |
EvaluatorPointerType | output_accessor |
const Index | preserve_elements_num_groups |
const Index | reduce_elements_num_groups |
ScratchAcc | scratch |
Definition at line 253 of file TensorReductionSycl.h.
typedef Evaluator::CoeffReturnType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::CoeffReturnType |
Definition at line 254 of file TensorReductionSycl.h.
typedef Evaluator::EvaluatorPointerType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::EvaluatorPointerType |
Definition at line 255 of file TensorReductionSycl.h.
typedef Evaluator::Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::Index |
Definition at line 256 of file TensorReductionSycl.h.
typedef OpDef::type Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::Op |
Definition at line 258 of file TensorReductionSycl.h.
typedef OpDefiner<OpType, CoeffReturnType, Index, false> Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::OpDef |
Definition at line 257 of file TensorReductionSycl.h.
typedef cl::sycl::accessor<CoeffReturnType, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local> Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::ScratchAcc |
Definition at line 260 of file TensorReductionSycl.h.
|
inline |
Definition at line 270 of file TensorReductionSycl.h.
|
inline |
Definition at line 282 of file TensorReductionSycl.h.
|
inline |
Definition at line 300 of file TensorReductionSycl.h.
Evaluator Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::evaluator |
Definition at line 262 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::num_coeffs_to_preserve |
Definition at line 267 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::num_coeffs_to_reduce |
Definition at line 268 of file TensorReductionSycl.h.
Op Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::op |
Definition at line 264 of file TensorReductionSycl.h.
EvaluatorPointerType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::output_accessor |
Definition at line 263 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::preserve_elements_num_groups |
Definition at line 265 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::reduce_elements_num_groups |
Definition at line 266 of file TensorReductionSycl.h.
ScratchAcc Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::scratch |
Definition at line 261 of file TensorReductionSycl.h.