torch_sys::c_generated

Function atg_quantized_gru_cell

source
pub unsafe extern "C" fn atg_quantized_gru_cell(
    out__: *mut *mut C_tensor,
    input_: *mut C_tensor,
    hx_: *mut C_tensor,
    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,
)