device_multiple_reduce.hpp File Reference#
device_multiple_reduce.hpp File Reference
#include <vector>#include <memory>#include <array>#include <iostream>#include "ck/ck.hpp"#include "ck/tensor_operation/gpu/device/device_base.hpp"#include "ck/utility/reduction_enums.hpp"Go to the source code of this file.
Classes | |
| struct | ck::tensor_operation::device::DeviceMultipleReduce< Rank, NumReduceDim, NumReduction, InElementwiseOperationTuple, AccElementwiseOperationTuple > |
Namespaces | |
| namespace | ck |
| namespace | ck::tensor_operation |
| namespace | ck::tensor_operation::device |
Typedefs | |
| template<index_t Rank, index_t NumReduceDim, index_t NumReduction, typename InElementwiseOperationTuple, typename AccElementwiseOperationTuple> | |
| using | ck::tensor_operation::device::DeviceMultipleReducePtr |