pub unsafe extern "C" fn atg_diag_backward( out__: *mut *mut C_tensor, grad_: *mut C_tensor, input_sizes_data: *const i64, input_sizes_len: c_int, diagonal_: i64)