Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dattention.cu108 auto NH = q_k_v.size(2); in transform_bias_rescale_qkv_kernel() local
211 const auto NH = q_k_v.size(2); in transform_bias_rescale_qkv_add_padding_kernel() local