pub unsafe extern "C" fn atg__transform_bias_rescale_qkv( out__: *mut *mut C_tensor, qkv_: *mut C_tensor, qkv_bias_: *mut C_tensor, num_heads_: i64, )