Home
last modified time | relevance | path

Searched defs:b_staging (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
Dmma_from_smem.h470 typename Base::TensorRefB in MmaPipelinedFromSharedMemory()
502 typename Base::TensorRefB b_staging, ///< staging memory for loading B in MmaPipelinedFromSharedMemory()