pub unsafe extern "C" fn atg_lstm_cell(
out__: *mut *mut C_tensor,
input_: *mut C_tensor,
hx_data: *const *mut C_tensor,
hx_len: c_int,
w_ih_: *mut C_tensor,
w_hh_: *mut C_tensor,
b_ih_: *mut C_tensor,
b_hh_: *mut C_tensor,
)