torch_sys::c_generated

Function atg_mkldnn_rnn_layer

source
pub unsafe extern "C" fn atg_mkldnn_rnn_layer(
    out__: *mut *mut C_tensor,
    input_: *mut C_tensor,
    weight0_: *mut C_tensor,
    weight1_: *mut C_tensor,
    weight2_: *mut C_tensor,
    weight3_: *mut C_tensor,
    hx__: *mut C_tensor,
    cx__: *mut C_tensor,
    reverse_: c_int,
    batch_sizes_data: *const i64,
    batch_sizes_len: c_int,
    mode_: i64,
    hidden_size_: i64,
    num_layers_: i64,
    has_biases_: c_int,
    bidirectional_: c_int,
    batch_first_: c_int,
    train_: c_int,
)