pub unsafe extern "C" fn atg_cosine_similarity( out__: *mut *mut C_tensor, x1_: *mut C_tensor, x2_: *mut C_tensor, dim_: i64, eps_: f64, )