torch_sys::c_generated

Function atg_quantized_lstm_cell

source
pub unsafe extern "C" fn atg_quantized_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,
    packed_ih_: *mut C_tensor,
    packed_hh_: *mut C_tensor,
    col_offsets_ih_: *mut C_tensor,
    col_offsets_hh_: *mut C_tensor,
    scale_ih_: *mut C_scalar,
    scale_hh_: *mut C_scalar,
    zero_point_ih_: *mut C_scalar,
    zero_point_hh_: *mut C_scalar,
)