Member List#
ck_tile::BlockDropout Member List
This is the complete list of members for ck_tile::BlockDropout, including all inherited members.
BlockDropout(index_t i_batch, index_t i_head, index_t nheads, unsigned long long seed, unsigned long long offset, float rp_undrop_, uint8_t p_undrop_in_uint8_t_, bool is_store_randval_) | ck_tile::BlockDropout | inline |
is_store_randval | ck_tile::BlockDropout | |
MakeRandvalDramWindow(RandValDramBlockWindowTmp &randval_dram_block_window_tmp, index_t seqlen_qk_start) | ck_tile::BlockDropout | inlinestatic |
MakeRandValLdsBlockDescriptor() | ck_tile::BlockDropout | inlinestatic |
MakeRandValLdsShuffleTileDistribution() | ck_tile::BlockDropout | inlinestatic |
MakeRandValTileDistribution() | ck_tile::BlockDropout | inlinestatic |
p_undrop_in_uint8_t | ck_tile::BlockDropout | |
ph | ck_tile::BlockDropout | |
rp_undrop | ck_tile::BlockDropout | |
Run(void *randval_ptr, const index_t start_n0_idx, PComputeWindow &p_compute, RandValDramWindow &randval_dram_window) const | ck_tile::BlockDropout | inline |