Searched defs:NH (Results 1 – 1 of 1) sorted by relevance
108 auto NH = q_k_v.size(2); in transform_bias_rescale_qkv_kernel() local211 const auto NH = q_k_v.size(2); in transform_bias_rescale_qkv_add_padding_kernel() local