TopkSoftmaxHostArgs Struct Reference

TopkSoftmaxHostArgs Struct Reference#

Composable Kernel: ck_tile::TopkSoftmaxHostArgs Struct Reference
Public Attributes | List of all members
ck_tile::TopkSoftmaxHostArgs 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

index_t ck_tile::TopkSoftmaxHostArgs::num_experts

◆ num_rows

index_t ck_tile::TopkSoftmaxHostArgs::num_rows

◆ p_indices

void* ck_tile::TopkSoftmaxHostArgs::p_indices

◆ p_input

const void* ck_tile::TopkSoftmaxHostArgs::p_input

◆ p_output

void* ck_tile::TopkSoftmaxHostArgs::p_output

◆ stride_input

index_t ck_tile::TopkSoftmaxHostArgs::stride_input

◆ stride_output

index_t ck_tile::TopkSoftmaxHostArgs::stride_output

◆ topk

index_t ck_tile::TopkSoftmaxHostArgs::topk
The documentation for this struct was generated from the following file: