TopkSoftmaxKargs Struct Reference

TopkSoftmaxKargs Struct Reference#

Composable Kernel: ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs Struct Reference
Public Attributes | List of all members
ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs Struct Reference

#include <topk_softmax_kernel.hpp>

Public Attributes

const void * p_input
void * p_output
void * p_indices
index_t num_rows
index_t num_experts
index_t topk
index_t stride_input
index_t stride_output

Member Data Documentation

◆ num_experts

template<typename Pipeline_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::num_experts

◆ num_rows

template<typename Pipeline_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::num_rows

◆ p_indices

template<typename Pipeline_>
void* ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::p_indices

◆ p_input

template<typename Pipeline_>
const void* ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::p_input

◆ p_output

template<typename Pipeline_>
void* ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::p_output

◆ stride_input

template<typename Pipeline_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::stride_input

◆ stride_output

template<typename Pipeline_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::stride_output

◆ topk

template<typename Pipeline_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::TopkSoftmaxKargs::topk
The documentation for this struct was generated from the following file: