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:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/topk_softmax/kernel/topk_softmax_kernel.hpp