Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/warp/
H A Dmma_tensorop_dequantizer.h146 void load(FragmentScale& scale_frag) in load()
156 void dequantize(FragmentDequantizedOperand& operand_frag, const FragmentScale& scale_frag) in dequantize()
254 void load(FragmentScale& scale_frag) in load()
264 void dequantize(FragmentDequantizedOperand& operand_frag, const FragmentScale& scale_frag) in dequantize()
348 void load(FragmentScale& scale_frag) in load()
360 void dequantize(FragmentDequantizedOperand& operand_frag, const FragmentScale& scale_frag) in dequantize()
434 void load(FragmentScale& scale_frag) in load()
449 void dequantize(FragmentDequantizedOperand& operand_frag, const FragmentScale& scale_frag) in dequantize()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
H A Dmma_from_smem.h312 static Fragment apply(Fragment frag, FragmentScale const& scale_frag) { in apply()