Searched defs:gK (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | flash_fwd_kernel.h | 130 … Tensor gK = local_tile(mK(_, bidh / params.h_h_k_ratio, _), Shape<Int<kBlockN>, Int<kHeadDim>>{}, in compute_attn_1rowblock() local
582 Tensor gK = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.k_ptr) + row_offset_k), in compute_attn_1rowblock_splitkv() local
|
| D | flash_bwd_kernel.h | 131 Tensor gK = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.k_ptr) + row_offset_k), in compute_dq_dk_dv_1colblock() local
|
| /external/chromium-trace/catapult/common/py_vulcanize/third_party/rjsmin/bench/ |
| D | apiviewer.js | 175 …",b=".prototype",L="function",K="Boolean",J="Error",I="constructor",H="warn",G="hasOwnProperty",F=…
|