Deep Neural Network Library (DNNL)  1.1.3
Performance library for Deep Learning
Public Member Functions | List of all members
dnnl::vanilla_rnn_forward::desc Struct Reference

Descriptor for RNN forward propagation. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::vanilla_rnn_forward::desc:
Collaboration graph
[legend]

Public Member Functions

 desc (prop_kind aprop_kind, algorithm activation, rnn_direction direction, const memory::desc &src_layer_desc, const memory::desc &src_iter_desc, const memory::desc &weights_layer_desc, const memory::desc &weights_iter_desc, const memory::desc &bias_desc, const memory::desc &dst_layer_desc, const memory::desc &dst_iter_desc, rnn_flags flags=rnn_flags::undef, float alpha=0.0f, float beta=0.0f)
 Initializes an RNN descriptor for forward propagation using prop_kind, activation, direction, and memory descriptors. More...
 

Detailed Description

Descriptor for RNN forward propagation.

Constructor & Destructor Documentation

◆ desc()

dnnl::vanilla_rnn_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  activation,
rnn_direction  direction,
const memory::desc &  src_layer_desc,
const memory::desc &  src_iter_desc,
const memory::desc &  weights_layer_desc,
const memory::desc &  weights_iter_desc,
const memory::desc &  bias_desc,
const memory::desc &  dst_layer_desc,
const memory::desc &  dst_iter_desc,
rnn_flags  flags = rnn_flags::undef,
float  alpha = 0.0f,
float  beta = 0.0f 
)
inline

Initializes an RNN descriptor for forward propagation using prop_kind, activation, direction, and memory descriptors.

Note
If prop_kind equals dnnl::forward_training, you must query a workspace memory descriptor before creating the primitive.

alpha, beta and flags are parameters to the RNN descriptor. If activation is eltwise_relu, alpha represents the negative slope. beta and flags are currently ignored.

src_iter_desc, bias_desc, and dst_iter_desc are allowed to point to a zero memory descriptor, which would indicate that the RNN primitive should not use them.

Note
All memory descriptors except src_iter_desc can be initialized with an dnnl::memory::format_tag::any value of format_kind.
The documentation for this struct was generated from the following file: