Skip to main content

build_mha_fwd_graph

atomr_accel_cuda::kernel::cudnn::attention

Function build_mha_fwd_graph 

Source 
pub fn build_mha_fwd_graph(
    dtype: DtypeTag,
    p: &AttentionParams,
    layout: TensorLayout,
) -> OperationGraphSpec