#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 292 of file TensorReductionSycl.h.
typedef Evaluator::CoeffReturnType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::CoeffReturnType |
Definition at line 293 of file TensorReductionSycl.h.
typedef Evaluator::EvaluatorPointerType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::EvaluatorPointerType |
Definition at line 294 of file TensorReductionSycl.h.
typedef Evaluator::Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::Index |
Definition at line 295 of file TensorReductionSycl.h.
typedef OpDef::type Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::Op |
Definition at line 297 of file TensorReductionSycl.h.
typedef OpDefiner<OpType, CoeffReturnType, Index, false> Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::OpDef |
Definition at line 296 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 299 of file TensorReductionSycl.h.
|
inline |
Definition at line 309 of file TensorReductionSycl.h.
|
inline |
Definition at line 321 of file TensorReductionSycl.h.
|
inline |
Definition at line 339 of file TensorReductionSycl.h.
Evaluator Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::evaluator |
Definition at line 301 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::num_coeffs_to_preserve |
Definition at line 306 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::num_coeffs_to_reduce |
Definition at line 307 of file TensorReductionSycl.h.
Op Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::op |
Definition at line 303 of file TensorReductionSycl.h.
EvaluatorPointerType Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::output_accessor |
Definition at line 302 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::preserve_elements_num_groups |
Definition at line 304 of file TensorReductionSycl.h.
const Index Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::reduce_elements_num_groups |
Definition at line 305 of file TensorReductionSycl.h.
ScratchAcc Eigen::TensorSycl::internal::PartialReductionKernel< Evaluator, OpType, PannelParameters, rt >::scratch |
Definition at line 300 of file TensorReductionSycl.h.