[][src]Function onednn_sys::dnnl_lbr_gru_forward_desc_init

pub unsafe extern "C" fn dnnl_lbr_gru_forward_desc_init(
    rnn_desc: *mut dnnl_rnn_desc_t,
    prop_kind: dnnl_prop_kind_t,
    direction: dnnl_rnn_direction_t,
    src_layer_desc: *const dnnl_memory_desc_t,
    src_iter_desc: *const dnnl_memory_desc_t,
    weights_layer_desc: *const dnnl_memory_desc_t,
    weights_iter_desc: *const dnnl_memory_desc_t,
    bias_desc: *const dnnl_memory_desc_t,
    dst_layer_desc: *const dnnl_memory_desc_t,
    dst_iter_desc: *const dnnl_memory_desc_t,
    flags: c_uint
) -> dnnl_status_t

Initializes a descriptor for LBR GRU forward propagation primitive.

The following arguments may either be @c NULL or point to a zero memory descriptor:

  • @p src_iter_desc,
  • @p bias_desc,
  • @p dst_iter_desc.

This would then indicate that the LBR GRU forward propagation primitive should not use them and should default to zero values instead.

@param rnn_desc Output descriptor for LBR GRU primitive. @param prop_kind Propagation kind. Possible values are #dnnl_forward_training and #dnnl_forward_inference. @param direction RNN direction. See @ref dnnl_rnn_direction_t for more info. @param src_layer_desc Memory descriptor for the input vector. @param src_iter_desc Memory descriptor for the input recurrent hidden state vector. @param weights_layer_desc Memory descriptor for the weights applied to the layer input. @param weights_iter_desc Memory descriptor for the weights applied to the recurrent input. @param bias_desc Bias memory descriptor. @param dst_layer_desc Memory descriptor for the output vector. @param dst_iter_desc Memory descriptor for the output recurrent hidden state vector. @param flags Unused. @returns #dnnl_success on success and a status describing the error otherwise.