torch_sys::c_generated

Function atg_lstm_cell

source
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,
)