BlockSoftmax2D< Problem_, Policy_ > Struct Template Reference#
ck_tile::BlockSoftmax2D< Problem_, Policy_ > Struct Template Reference
#include <block_softmax_2d.hpp>
Public Types | |
using | Problem = remove_cvref_t< Problem_ > |
using | Policy = remove_cvref_t< Policy_ > |
using | DataType = typename Problem::DataType |
Public Member Functions | |
template<typename DistributedTensor , index_t dim = 1> | |
CK_TILE_DEVICE void | operator() (const DistributedTensor &x, DistributedTensor &y, number< dim >={}) |
template<typename DistributedTensor , index_t dim = 1> | |
decltype(auto) CK_TILE_DEVICE | operator() (const DistributedTensor &x, number< dim >={}) |
Member Typedef Documentation
◆ DataType
template<typename Problem_ , typename Policy_ = void>
using ck_tile::BlockSoftmax2D< Problem_, Policy_ >::DataType = typename Problem::DataType |
◆ Policy
template<typename Problem_ , typename Policy_ = void>
using ck_tile::BlockSoftmax2D< Problem_, Policy_ >::Policy = remove_cvref_t<Policy_> |
◆ Problem
template<typename Problem_ , typename Policy_ = void>
using ck_tile::BlockSoftmax2D< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_> |
Member Function Documentation
◆ operator()() [1/2]
template<typename Problem_ , typename Policy_ = void>
template<typename DistributedTensor , index_t dim = 1>
|
inline |
◆ operator()() [2/2]
template<typename Problem_ , typename Policy_ = void>
template<typename DistributedTensor , index_t dim = 1>
|
inline |
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/softmax/block/block_softmax_2d.hpp