torch_sys::c_generated

Function atg__cudnn_rnn_out

source
pub unsafe extern "C" fn atg__cudnn_rnn_out(
    out__: *mut *mut C_tensor,
    out0_: *mut C_tensor,
    out1_: *mut C_tensor,
    out2_: *mut C_tensor,
    out3_: *mut C_tensor,
    out4_: *mut C_tensor,
    input_: *mut C_tensor,
    weight_data: *const *mut C_tensor,
    weight_len: c_int,
    weight_stride0_: i64,
    weight_buf_: *mut C_tensor,
    hx_: *mut C_tensor,
    cx_: *mut C_tensor,
    mode_: i64,
    hidden_size_: i64,
    proj_size_: i64,
    num_layers_: i64,
    batch_first_: c_int,
    dropout_: f64,
    train_: c_int,
    bidirectional_: c_int,
    batch_sizes_data: *const i64,
    batch_sizes_len: c_int,
    dropout_state_: *mut C_tensor,
)