BlockSoftmax2D< Problem_, Policy_ > Struct Template Reference

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

Composable Kernel: ck_tile::BlockSoftmax2D< Problem_, Policy_ > Struct Template Reference
Public Types | Public Member Functions | List of all members
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>
CK_TILE_DEVICE decltype(auto) 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>
CK_TILE_DEVICE decltype(auto) ck_tile::BlockSoftmax2D< Problem_, Policy_ >::operator() ( const DistributedTensor & x,
number< dim > = {} )
inline
The documentation for this struct was generated from the following file: