pub unsafe extern "C" fn atg_cosine_embedding_loss( out__: *mut *mut C_tensor, input1_: *mut C_tensor, input2_: *mut C_tensor, target_: *mut C_tensor, margin_: f64, reduction_: i64, )