TopkSoftmaxWarpPerRowPolicy Struct Reference

TopkSoftmaxWarpPerRowPolicy Struct Reference#

Composable Kernel: ck_tile::TopkSoftmaxWarpPerRowPolicy Struct Reference
Static Public Member Functions | List of all members
ck_tile::TopkSoftmaxWarpPerRowPolicy Struct Reference

#include <topk_softmax_warp_per_row_policy.hpp>

Static Public Member Functions

template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto MakeInputDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto MakeOutputDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetSoftmax ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetTopk ()

Member Function Documentation

◆ GetSoftmax()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::TopkSoftmaxWarpPerRowPolicy::GetSoftmax ( )
inlinestaticconstexpr

◆ GetTopk()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::TopkSoftmaxWarpPerRowPolicy::GetTopk ( )
inlinestaticconstexpr

◆ MakeInputDistribution()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::TopkSoftmaxWarpPerRowPolicy::MakeInputDistribution ( )
inlinestaticconstexpr

◆ MakeOutputDistribution()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::TopkSoftmaxWarpPerRowPolicy::MakeOutputDistribution ( )
inlinestaticconstexpr
The documentation for this struct was generated from the following file: