BlockSoftmax2D< Problem_, Policy_ > Struct Template Reference

BlockSoftmax2D&lt; Problem_, Policy_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::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>
CK_TILE_DEVICE void ck_tile::BlockSoftmax2D< Problem_, Policy_ >::operator() ( const DistributedTensor &  x,
DistributedTensor &  y,
number< dim >  = {} 
)
inline

◆ operator()() [2/2]

template<typename Problem_ , typename Policy_ = void>
template<typename DistributedTensor , index_t dim = 1>
decltype(auto) CK_TILE_DEVICE ck_tile::BlockSoftmax2D< Problem_, Policy_ >::operator() ( const DistributedTensor &  x,
number< dim >  = {} 
)
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