Index

A | F | R | S

A

allocated_fused_rms_norm_qkv (in module nki_samples.reference.allocated_fused_linear)

allocated_fused_self_attn_for_SD_small_head_size (in module nki_samples.reference.allocated_attention)

F

flash_attn_bwd (in module nki_samples.reference.attention)

R

resize_nearest_fixed_dma_kernel (in module nki_samples.reference.vision)

S

select_and_scatter_kernel (in module nki_samples.reference.vision)