/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/device_reduce_multi_d.hpp Source File#
device_reduce_multi_d.hpp
Go to the documentation of this file.
std::unique_ptr< DeviceReduceMultiD< InDataType, DsDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, OutElementwiseOperation > > DeviceReduceMultiDPtr
Definition: device_reduce_multi_d.hpp:65
Definition: ck.hpp:267
Definition: device_base.hpp:77
Definition: device_reduce_multi_d.hpp:26
static constexpr index_t NumOutDim
Definition: device_reduce_multi_d.hpp:27
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer(const std::array< index_t, Rank > inLengths, const std::array< index_t, Rank > inStrides, const std::array< std::array< index_t, NumOutDim >, NumDTensor > DsLengths, const std::array< std::array< index_t, NumOutDim >, NumDTensor > DsStrides, const std::array< index_t, NumOutDim > outLengths, const std::array< index_t, NumOutDim > outStrides, const std::array< int, NumReduceDim > reduceDims, const void *in_dev, const std::array< const void *, NumDTensor > ds_dev, void *out_dev, const InElementwiseOperation in_elementwise_op, const OutElementwiseOperation out_elementwise_op)=0
static constexpr index_t NumDTensor
Definition: device_reduce_multi_d.hpp:29
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer()=0