torch_sys::c_generated

Function atg_lstm_mps_backward

source
pub unsafe extern "C" fn atg_lstm_mps_backward(
    out0_: *mut C_tensor,
    out1_data: *const *mut C_tensor,
    out1_len: c_int,
    out2_data: *const *mut C_tensor,
    out2_len: c_int,
    grad_y_: *mut C_tensor,
    grad_hy_: *mut C_tensor,
    grad_cy_: *mut C_tensor,
    z_state_: *mut C_tensor,
    cell_state_fwd_: *mut C_tensor,
    input_: *mut C_tensor,
    layersOutputs_: *mut C_tensor,
    hx_data: *const *mut C_tensor,
    hx_len: c_int,
    params_data: *const *mut C_tensor,
    params_len: c_int,
    has_biases_: c_int,
    num_layers_: i64,
    dropout_: f64,
    train_: c_int,
    bidirectional_: c_int,
    batch_first_: c_int,
)