/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/device_multiple_reduce.hpp Source File#
device_multiple_reduce.hpp
Go to the documentation of this file.
std::unique_ptr< DeviceMultipleReduce< Rank, NumReduceDim, NumReduction, InElementwiseOperationTuple, AccElementwiseOperationTuple > > DeviceMultipleReducePtr
Definition: device_multiple_reduce.hpp:54
Definition: ck.hpp:264
Definition: device_base.hpp:76
Definition: device_multiple_reduce.hpp:25
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer(const std::array< index_t, NumInputDim > inLengths, const std::array< index_t, NumInputDim > inStrides, const std::array< index_t, NumOutputDim > outLengths, const std::array< std::array< index_t, NumOutputDim >, NumReduction > outStrides, const std::array< int, NumReduceDim > reduceDims, const std::array< double, NumReduction > alphas, const std::array< double, NumReduction > betas, const void *in_dev, const std::array< void *, NumReduction > out_dev_buffers, const InElementwiseOperationTuple in_elementwise_op_tuple, const AccElementwiseOperationTuple acc_elementwise_op_tuple)=0
static constexpr index_t NumInputDim
Definition: device_multiple_reduce.hpp:26
static constexpr index_t NumOutputDim
Definition: device_multiple_reduce.hpp:27
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer()=0