Searched defs:inv_sqrt_dim_per_head (Results 1 – 1 of 1) sorted by relevance
105 const scalar_t inv_sqrt_dim_per_head) { in transform_bias_rescale_qkv_kernel()208 const scalar_t inv_sqrt_dim_per_head) { in transform_bias_rescale_qkv_add_padding_kernel()