/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/host/reference/reference_reduce.hpp File Reference#
reference_reduce.hpp File Reference
Go to the source code of this file.
Namespaces | |
ck_tile | |
Functions | |
template<typename XDataType , typename ComputeDataType , typename YDataType , typename ReduceOp > | |
CK_TILE_HOST void | ck_tile::reference_reduce (const HostTensor< XDataType > &x_m_n, HostTensor< YDataType > &y_m, ReduceOp reduce_op) |
template<typename XDataType , typename ComputeDataType , typename YDataType , typename ReduceOp , typename KeptDim , typename ReduceDims > | |
CK_TILE_HOST void | ck_tile::reference_reduce (const HostTensor< XDataType > &x_tensor, HostTensor< YDataType > &y_tensor, ReduceOp reduce_op, KeptDim kept_dim, ReduceDims reduce_dims) |