torch_sys::c_generated

Function atg__cudnn_rnn_flatten_weight

source
pub unsafe extern "C" fn atg__cudnn_rnn_flatten_weight(
    out__: *mut *mut C_tensor,
    weight_arr_data: *const *mut C_tensor,
    weight_arr_len: c_int,
    weight_stride0_: i64,
    input_size_: i64,
    mode_: i64,
    hidden_size_: i64,
    proj_size_: i64,
    num_layers_: i64,
    batch_first_: c_int,
    bidirectional_: c_int,
)